How can I use this model on my own data set?

[jlhou]

Thank you for doing such an outstanding job!How can I use this model on my own data set?

[speedinghzl]

Hi @jlhou, thanks for your interest in our work. There are several steps needed for training on the custom dataset:

1. prepare images and labels, and generate train and val files. (see examples)
2. modify the dataset.py to adapt your dataset.
3. run the script. need to add the parameter --num-classes and modify --data-dir --data-list

P.S. Only step 1 is required If you just want to run inference on your dataset.

[jlhou]

Thank you!

[LeeThrzz]

@jlhou hello, I meet the same problem as you, I have a 2 classes dataset to train, and I modify the dataset.py too, but I can't get the results, If you succeed, could you tell me how to correct it? thanks a lot!