search

YudeWang / semantic-segmentation-codebase

Codebase for semantic segmentation experiments
MIT License
57 stars 11 forks source link

the test on pretrained model is pretty good but those output imgs are black #5

Closed FUTUREEEEEE closed 3 years ago

FUTUREEEEEE commented 3 years ago

image the test results looks very good but the output imgs are all black (most of the output img are small ,about 2k each)

image should results looks like this? or i need to post-process them?

image

YudeWang commented 3 years ago

Hi @FUTUREEEEEE , Here is a function for visualization. https://github.com/YudeWang/semantic-segmentation-codebase/blob/995b8fa9d6b08e3ff1f15a7aef2ce5445deaa834/lib/datasets/VOCDataset.py#L115