pytorch-segmentation
pytorch-segmentation copied to clipboard
How to load and test the pre-trained segmentation models on system using python script?
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 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.