search

NVIDIA-AI-IOT / deepstream_tao_apps

Sample apps to demonstrate how to deploy models trained with TAO on DeepStream
MIT License
376 stars 96 forks source link

How to Use Mobilenet_v1 backbone with SSD #15

Open sparsh-b opened 4 years ago

sparsh-b commented 4 years ago

The model file provided for SSD uses resnet-18 as backbone. I want to use Mobilenet_v1 as backbone. Please let me know how to do the same.

mchi-zg commented 4 years ago

you need to use TLT to re-tranin the model, please refer to tlt doc https://developer.nvidia.com/transfer-learning-toolkit or ask questions in tlt forum - https://forums.developer.nvidia.com/c/accelerated-computing/intelligent-video-analytics/transfer-learning-toolkit/17