python使用opencv将Labelme生成的分割图转化成二值化分割图

2022.4.20补充

实力分割图是使用Labelme标注软件标注完成后，使用labelme代码转换生成的

代码去GitHub搜labelme下载就行

---

说明：使用labelme标注好车道线之后生成‘“.json”注释文件，运行labelme2voc.py生成语义分割或者实例分割图，如下图所示：


此时使用opencv将这些图转化成二值化分割图，代码如下：

import cv2
import os


def convert_png2seg(input_dir,output_dir):
    files = os.listdir(input_dir)
    for file in files:
        part = file.split('.')
        part0 = part[0]
        part1 = part[1]
        img = cv2.imread(input_dir+file)
        img_gray = cv2.cvtColor(img, cv2.COLOR_RGB2GRAY)

        retval, binary = cv2.threshold(img_gray, 10, 255, cv2.THRESH_BINARY)  #阈值处理
        cv2.imwrite(output_dir + part0 + '.' + part1, binary)


if __name__ == "__main__":
    input_dir = r"D:\desktop\peng/"
    output_dir = r"D:\desktop\output/"
    convert_png2seg(input_dir,output_dir)