pytorch-segmentation icon indicating copy to clipboard operation
pytorch-segmentation copied to clipboard

How to load and test the pre-trained segmentation models on system using python script?

Open NISHANTSHRIVASTAV opened this issue 3 years ago • 1 comments

Hi @dusty-nv ,

I want to load and test pre-trained segmentation models such as FCN-ResNet18-MHP, FCN-ResNet18-DeepScene on my system with python script e.g. test.py before using on Jetson.

How can we achieve the same?

Please let me know if any further information w.r.t is needed.

Thanks!

NISHANTSHRIVASTAV avatar Oct 21 '21 13:10 NISHANTSHRIVASTAV

@NISHANTSHRIVASTAV did you get any input on your question. I want to use the a pre-trained fcn-resnet-18-deepscene model and add a class for performing segmentaion on solar panes in a field. I don't want to have to train from scratch when I run the train.py it data that fcn-resent-18 I get an error that says it is not in the data set.

gabegarz avatar Nov 26 '21 16:11 gabegarz