Can't make a prediction

deep-diver / Soccer-Ball-Detection-YOLOv2

YOLOv2 trained against custom dataset

118 stars 43 forks source link

Can't make a prediction #8

Open IhorKosovych opened 5 years ago

IhorKosovych commented 5 years ago

Thanks for your awesome code share!

I trained the model. The performance was good.

step 10 - loss 119.7686767578125 - moving ave loss 119.82690792130654
Finish 10 epoch(es)
Checkpoint at step 10
Parsing cfg/yolov2.cfg
Loading None ...
Finished in 5.650520324707031e-05s

And then I tried to predict another image. But after

print(tfnet.return_predict(original_img))

The result is:

[]

Can somebody help me?

veshitala commented 5 years ago

options = {"model": "cfg/yolov2-voc-1c.cfg", "load": 1000, "gpu": 1.0, "threshold":0.01}

reducing your thresold will help

Zrufy commented 4 years ago

at the end of the training have you some error?