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>
posted @ 2019-04-02 15:54  Le1B_o  阅读(1165)  评论(0编辑  收藏  举报