Train Custom image Segmentation Model jetson_inference

Hello, I am looking for a tutorial that can provide with necessary steps to train an image segmentation model.


Please check the tutorial below:


I found a file in the jetson_inference directory under jetson_inference/python/training/segmentation and I was wondering can this be used to train a custom segmentation model instead of nvidia-digits as digits is not python3 supported and also has been outdated. Thanks

Hi @nitinkumar96, the training code for the FCN-Resnet18 models in the repo is from here:

However it is resource intensive to train segmentation networks, and I trained those on a PC with a discrete GPU. There is also a tutorial a community member wrote here: