thtrieu / darkflow

Translate darknet to tensorflow. Load trained weights, retrain/fine-tune using tensorflow, export constant graph def to mobile devices
GNU General Public License v3.0
6.13k stars 2.08k forks source link

Bounding box is not right. #1040

Open HSHunterR opened 5 years ago

HSHunterR commented 5 years ago

I try on my dataset(360 images of 3 class objects),modify model tiny-yolo-4c, batch 8, use the default lr, gpu model(GTX 950M 2GB), about 9k steps(more than 10 hours), loss about 3 or 2, move ave loss the same, but the bounding box is not correct on train image( threshold is 0.0045); And then I use 6 images, train 150 epoch, batch 6, the loss is 3 or 2, then it can't more smaller, the bounding box is not correct too, how to solve it?

kmarrip commented 5 years ago

@HSHunterR loss is too high for any bounding box to be true , add more images to your dataset, for each class have at least 400 images, then retrain the model, make sure the loss is around 0.5

HSHunterR commented 5 years ago

Thanks for your suggestion!