HRNet / HRNet-Semantic-Segmentation

The OCR approach is rephrased as Segmentation Transformer: https://arxiv.org/abs/1909.11065. This is an official implementation of semantic segmentation for HRNet. https://arxiv.org/abs/1908.07919
Other
3.09k stars 682 forks source link

How to segment only selected classes (out of 19 in CityScapes) by using pretrained model? #273

Closed Arshadoid closed 1 year ago

Arshadoid commented 1 year ago

Hi, what part of the code do I need to change to do an inference during the testing on a single class?