做建筑物分割,用arcgis做出来的标签(tif)为纯黑,转为png格式后也是纯黑,虽然在arcgis中可以正常显示,且不影响模型训练,但是的确有那么点不好看,参考其他数据集,单类别表标签均为黑白二值图(0/255)。arcgis做的标签为纯黑的原因是他的像素背景是0,目标是0-1,因此肉眼看不出来,故将目标0-1重新赋值为255即可。
我是重新进行了二值化,用的标签格式为.png
import os
import cv2
def image_binarization(filepath,destpath):
pathDir = os.listdir(filepath)
for allDir in pathDir:
child = os.path.join(filepath,allDir)
dest = os.path.join(destpath,allDir)
if os.path.isfile(child):
image = cv2.imread(child)
gray = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
retval, dst = cv2.threshold(gray, 0, 255, cv2.THRESH_BINARY)
cv2.imwrite(dest ,dst)
if __name__ =="__main__":
filepath = r'../VOCData/11'
destpath = r'../VOCData/111'
image_binarization(filepath,destpath)
效果如下:
![](https://img-blog.csdnimg.cn/62ccb7022433407996f12f58d468c195.png)
![](https://img-blog.csdnimg.cn/abffef91a6b94ee19fbd49a7f246d69f.png)