experiencor / keras-yolo2

Easy training on custom dataset. Various backends (MobileNet and SqueezeNet) supported. A YOLO demo to detect raccoon run entirely in brower is accessible at https://git.io/vF7vI (not on Windows).
MIT License
1.73k stars 784 forks source link

Same bounding box for all images #403

Closed fenilsuchak closed 5 years ago

fenilsuchak commented 5 years ago

I'm training using the notebook Blood_Cell_Detection.ipynb on custom dataset. I'm also not using any pretrained weights. So after 7-8 epochs(initial 3 were warmup), average recall is around 0.75 and the train loss and val loss are continuously decreasing , but looks like its predicting the same bounding box for all the images. These are some of the train set images. It would be great if someone help me with this Thanks in advance

whatsapp image 2019-02-05 at 12 46 13 am whatsapp image 2019-02-05 at 12 46 10 am whatsapp image 2019-02-05 at 12 46 11 am