Train mobilenet v2 512?

Hello, I can train 512 following guides on here but it seems to be specific to the v1 base model - is there a v2 model I can download and use as the base for the 512 version?

many thanks


Would you mind sharing the guide you mentioned with us first?

Do you use jetson-inference?
If yes, you can modify the model data for MobileNet v2 to train directly.


Hi @liellplane, you can download other base models for here:

I haven’t tested SSD-Mobilenet-v2 with higher resolutions though - for starters, you would need to add config.set_image_size(args.resolution) under this line of code:

Thanks AastaLLL, its not really a guide it was various topics on here such as :
How train jetson-inference ssd512 model - Jetson & Embedded Systems / Jetson TX2 - NVIDIA Developer Forums

Thanks for that link! OK will start there - in your opinion is there much difference in performance between V1 and V2 on a Xavier to warrant the effort to use V2?

@liellplane, no in my basic testing of SSD-Mobilenet-v1 vs SSD-Mobilenet-v2 on Xavier NX, there were no real appreciable differences (SSD-Mobilenet-v2 was a couple percent slower actually), so I just stuck with using SSD-Mobilenet-v1 since it is very stable to train and deploy at this point.

very helpful info, thanks!!

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