Closed cschar closed 5 years ago
I'm having the same issue
@cschar retrain_yolo.py only generates a weights file. You then have to run yad2k.py with that weights file and it will generate a model that will work for test_yolo.py
Ah thanks @alexfaus08 !
python ./test_yolo.py ./trained_stage_3.h5 --classes_path ./datasets/set1/classes.txt
Error is as follows:
It can successfully train the model in retrain_yolo.py, and then draw boxes, but can't load up using the test_yolo.py command.
As of now i'm going to just hack retrain_yolo.py into another file to test with, but I would have liked to use the test_yolo.py directly.