Error when training coco @batch 1473, epoch 0 about pytorch-yolov3 HOT 4 CLOSED

Because img is a numpy array that wraps the returned PIL Image, I am guessing that None is returned, and so np.array(None).shape is (), hence the got 0.

from pytorch-yolov3.

@alexandru-dinu the training is fine from start, and the error happened @Batch 1473....i don't understand

from pytorch-yolov3.

I got similar issue in detect.py.
The attached data/samples can be run, however, if grayscale images are given as data/samples, then "not enough values to unpack" error is reproduced.
The code h,w,_=img.shape caused this error because the grayscale image does not have channels, i.e. , len(img.shape)==2.

I think the @Batch 1473 's images are broken (including None). The #36 issue may be caused by the dataset. Until @Batch 1472, the dataset is normal, I think.
My dirty hack is (1) get len(img.shape) and (2) use if divided; if len(img.shape)!=3, then this batch is ignored.

from pytorch-yolov3.

This was caused by a grayscale image in the dataset. It's fixed in master.

from pytorch-yolov3.