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

Can not predict using my trained weight #326

Open yuyijie1995 opened 6 years ago

yuyijie1995 commented 6 years ago

image When I used my trained weight to predict my image ,there was the error showing up. I don't know what is wrong . My config file like below. image How can I solve this problem? Thanks a lot

ilikedata commented 6 years ago

I was able to solve this error by using the required version of keras==2.0.8

alex-danilin commented 6 years ago

Having the same issue. Does anyone have a solution?