How to Get labelled Output

meetps / pytorch-semseg

Semantic Segmentation Architectures Implemented in PyTorch

https://meetshah.dev/semantic-segmentation/deep-learning/pytorch/visdom/2017/06/01/semantic-segmentation-over-the-years.html

MIT License

3.38k stars 799 forks source link

How to Get labelled Output #217

Open ZiqiFrancisFan opened 5 years ago

ZiqiFrancisFan commented 5 years ago

I am trying to get the labelled output using the trained model. The test.py only provides the RGB output. Is there an easy way to get the labelled outputs of test inputs using the trained model? Thanks a lot for your advice!