MS COCO 生成语义分割Png

最新推荐文章于 2025-07-03 15:50:18 发布

weixin_34205826

最新推荐文章于 2025-07-03 15:50:18 发布

阅读量414

点赞数

CC 4.0 BY-SA版权

文章标签：人工智能

原文链接：http://www.cnblogs.com/leebxo/p/10643110.html

本文介绍如何使用PythonAPI/cocoSegmentationToPngDemo.py函数进行语义分割，并针对2017数据集调整代码实现。解决在转换过程中遇到的ValueError: invalidpalettesize错误。

PythonAPI/cocoSegmentationToPngDemo.py函数是用来做语义分割的，参考这里https://blog.youkuaiyun.com/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

转载于:https://www.cnblogs.com/leebxo/p/10643110.html