MS COCO 生成语义分割Png
PythonAPI/cocoSegmentationToPngDemo.py函数是用来做语义分割的,参考这里https://blog.csdn.net/qq_33000225/article/details/78985635?utm_source=blogxgwz2
由于我用的是2017数据集,因此只需要改第一部分,labelMap = cocoSegmentationToSegmentationMap(coco, imgId, checkUniquePixelLabel=False, includeCrowd=includeCrowd)
不需要改第二部分。改完后去PythonAPI路径下make。
中途报错:ValueError: invalid palette size,
解决方案:
引用cmap的时候改成 np.uint8(cmap).tolist()
即可~
参考 https://stackoverflow.com/questions/50140853/pil-putpalette-error
</div>