Train detector different from SSD-Mobilenet on Jetson Nano

Autonomous Machines Jetson & Embedded Systems Jetson Nano
1

Good afternoon all,

I have successfully trained an SSD-Mobilenet on my custom dataset following the tutorial: jetson-inference/docs/pytorch-ssd.md at master · dusty-nv/jetson-inference · GitHub.
For comparison purposes, I would like to train different detectors on the same dataset. I can see that there are other pre-trained detectors available: jetson-inference/docs/detectnet-console-2.md at master · dusty-nv/jetson-inference · GitHub.
Is it possible to train these detectors via train_ssd.py by adding “–network=new_detector”? If not, how these pre-trained detectors can be trained on a custom dataset?

Regards,
Federico

3

Hi,

The following are TAO models.
Please find below to retrain a TAO model:
(Please note that you will need to run the toolkit on a desktop device)

Thanks.

5

This topic was automatically closed 14 days after the last reply. New replies are no longer allowed.