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.95k stars 1.1k forks source link

Align the index between pred and csv #288

Open Adolfhill opened 10 months ago

Adolfhill commented 10 months ago

the index of object150_info.csv start from 1, but pred start from 0. So it should be names[int(row[0]) - 1] instead of names[int(row[0])]