Retrain ssd mobilnet

Hello iam using jetson inference to train object detection model. I did the training for 60 epochs and its complete.
Now i want to add more pictures to the dataset and retrain the model.
Is there a way to retrain the model rather than starting training from scratch?
Thank you

Hi,

Yes. Please use the --pretrained-ssd configuration to input a pre-trained model.
EX.

$ python3 train_ssd.py --pretrained-ssd=models/mobilenet-v1-ssd-mp-0_675.pth ...

For more configure support, you can check this file directly.

Thanks.

1 Like