how to load a pretrained classification model weights and continue to train the detection model

[BoXiao123]

Hi, I have a mobilenet weights pretrained on ImageNet. I am wondering is there a way to train mobilenet-ssd initialized from this mobilenet weights. Thanks.

[edmBernard]

something like that should work :

train.py --network mobilenet --data-shape 224 --pretrained your_pretrained_model_prefix --epoch XXX --prefix output_model_prefix

you need to take care if your mobilenet is exactly the same as in ssd code.

[BoXiao123]

hi @edmBernard thx for your reply. It works