1. 程式人生 > >處理coco資料集-語義分割

處理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