How to train my own dataset?

[CRonaldo1997]

Hi thanks for your work, how can I train the model using own dataset？ Thanks

[qfgaohao]

Hi @CRonaldo1997 , I think the easiest way is as follows:

1. Convert you datasets into Pascal VOC format.
2. run "python train_ssd.py --datasets your-train-dataset-path --validation_dataset your-validation-dataset-path --net mobilenet-v1-ssd --base_net models/mobilenet_v1_with_relu_69_5.pth --batch_size 24 --num_epochs 200 --scheduler cosine --lr 0.01 --t_max 200"

Feel free to change the parameters.

[ramanathan831]

Hi @CRonaldo1997, I am not able to download the pre-trained model, can you upload in onedrive or similar to it and share the link here

[qfgaohao]

It's ok now. Plz refer to Issue #4

[kurianbenoy]

Hi @CRonaldo1997 , I think the easiest way is as follows:

1. Convert you datasets into Pascal VOC format.

For a dataset without Pascal VOOC format, removing years from data loader will work out right @qfgaohao ?