yolov3-tf2 icon indicating copy to clipboard operation
yolov3-tf2 copied to clipboard

Cannot detect anything

Open 1726400669 opened this issue 4 years ago • 3 comments

After I made my dataset and trained for 50 epochs with 1 class, I couldn't detect anything. I have changed the classes of detect.py to mine, weights to ./checkpoints/yolov3_train_50.tf, image to mine, and num_classes to 1, but the detection results are all 0 and lots of Warning like https://github.com/zzh8829/yolov3-tf2/issues/108 .

This is where I modify train.py BaiduShurufa_2019-12-15_11-56-51

and This is where I modify model.py: BaiduShurufa_2019-12-15_11-55-15

Thanks.

1726400669 avatar Dec 15 '19 03:12 1726400669

have you tried training using eager mode does it work?

zzh8829 avatar Dec 18 '19 09:12 zzh8829

I made a tutorial on training from scratch. https://github.com/zzh8829/yolov3-tf2/blob/master/docs/training_voc.md let me know if this helps you, in my experience training without transfer learning from darknet is almost impossible

zzh8829 avatar Dec 21 '19 07:12 zzh8829

Hi everyone, I used the instructions on the tutorial in this repo to train it on my custom pictures. I converted the darknet weight using convert.py and then used the eager_fit mode and transfer learning using this command: python train.py --dataset ./data/Ortho_Dataset/ortho_train.tfrecord --val_dataset ./data/Ortho_Dataset/ortho_valid.tfrecord --classes ./data/obj.names --num_classes 1 --mode eager_fit --transfer darknet --batch_size 16 --epochs 10 --weights ./checkpoints/yolov3.tf --weights_num_classes 80

The training results are shown in this picture.

Screen Shot 2021-06-29 at 9 18 27 AM

Although I donot encounter any problem when doing my taining, the model cannot detect anything from yolov3 and it returns zero and no detections. Refer to the following pic: Screen Shot 2021-06-29 at 9 18 41 AM

Does anyone have any idea what could be the problem?

Thanks,

kalikhademi avatar Jun 29 '21 16:06 kalikhademi