处理coco数据集-语义分割

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

猜你喜欢

转载自blog.csdn.net/aaon22357/article/details/83098456