Open PuneetKohli opened 6 years ago
retrain_yolo.py
does not have the ability to exclude specific categories from training. You will need to modify it or create a custom dataset with those classes excluded. A good solution might be to create a custom data loading function that only loads data into memory a few pieces at a time, and only selects the categories that you want to use.
retrain_yolo.py
assumes the input model is the one with default pretrained weights from darknet. To save your progress and continue training on a new dataset, you will have to modify the script.
Sorry, I have not had too much time to work on expanding functionality, and the repo owner seems to be inactive for now @allanzelener
Hi @shadySource Thanks for your reply. I don't think I am still clear based on your comments.
@PuneetKohli I am also trying to create a new dataset for it to train on, but I don't fully understand how retrain_yolo.py
expects the training data. How do I feed my images and my annotation files into retrainyolo ? Should the images and annotation xml files be in 2 different folders and should corresponding images and xml files have the same name? Also retrainyolo.py
expects a data_path
, classes_path
and anchors_path
. How do these correspond to my images and my annotation xmls?
Hey guys,
I've got the retraining script to work on one of my custom datasets.
My requirements are as follows-
Now, I have a few questions-
Let me know if my question is still not clear.. Thank you!