I've been trying to train a slimmer model but anytime I run the training command (ex: ./flow --model cfg/tiny-yolo-voc.cfg --load bin/tiny-yolo-voc.weights --train --database VOC2012/JPEGImages/ --annotation VOC2012/Annotations/ --batch 16) I get this error:
File "./flow", line 56, in <module>
print('Enter training ...'); tfnet.train()
File "/home/ubuntu/darkflow/net/flow.py", line 37, in train
for i, (x_batch, datum) in enumerate(batches):
File "/home/ubuntu/darkflow/net/yolo/data.py", line 146, in shuffle
inp, new_feed = self._batch(train_instance)
File "/home/ubuntu/darkflow/net/yolov2/data.py", line 27, in _batch
img = self.preprocess(path, allobj)
File "/home/ubuntu/darkflow/net/yolo/test.py", line 61, in preprocess
result = imcv2_affine_trans(im)
File "/home/ubuntu/darkflow/utils/im_transform.py", line 19, in imcv2_affine_trans
h, w, c = im.shape
AttributeError: 'NoneType' object has no attribute 'shape'
Right after parsing the dataset. Not really sure why this is happening.
please refer to
Try to delete history file in folder net/YOLO/parse-history.txt
Most helpful comment
please refer to
128
145