CSAILVision / semantic-segmentation-pytorch

Pytorch implementation for Semantic Segmentation/Scene Parsing on MIT ADE20K dataset
http://sceneparsing.csail.mit.edu/
BSD 3-Clause "New" or "Revised" License
4.96k stars 1.1k forks source link

Mismatch between result message and result image #263

Open yes89929 opened 3 years ago

yes89929 commented 3 years ago

I made custom dataset that has 3 classes.

And, I trained hrnet model with pretrained weights. python train.py --gpus 0,1,2 --cfg config/ade20k-hrnetv2.yaml

Then, I validated a one image used training.

Output message is Mean IoU: 1.0000, Accuracy: 100.00% image

But, when looking result image, it's not. image

I printed number of pixcel in intersection and union. And, number of gray pixcel is 12614. However, it is must be > 1,000,000, considering result image.

image ### intersection: [12614 4295 3970] union [12614 4295 3970]

I have no idea what is problem.