Possible bug in the CarvanaClassifier.predict() method
ceshine opened this issue · comments
First of all, thanks for sharing this well written solution!
For an image of size (N, N). The following line of code seems to return an array of size (N,), but the desired effect is to classify every pixel in that image, that is, an array of size(N, N):
carvana-challenge/src/nn/classifier.py
Line 197 in 517c2e5
So maybe change it to pred_arr = pred.data.cpu().numpy()
would make more sense.
(I use pytorch 0.2.0.post2)
Oh man thank you for this. I was chasing this bug for hours... Here is the commit which fix the bug. Such a stupid mistake of mine...
Cool. Glad to be of help.