彩色二值图保存为黑白二值图像

import os
import cv2

def convert_to_binary(input_folder, output_folder):
    # 遍历输入文件夹中的所有文件
    for file_name in os.listdir(input_folder):
        # 构建输入文件的完整路径
        input_path = os.path.join(input_folder, file_name)
        
        # 读取输入图像
        image = cv2.imread(input_path, cv2.IMREAD_GRAYSCALE)
        
        # 将图像二值化
        _, binary_image = cv2.threshold(image, 0, 255, cv2.THRESH_BINARY)
        
        # 构建输出文件的完整路径
        output_path = os.path.join(output_folder, file_name)
        
        # 保存二值图像
        cv2.imwrite(output_path, binary_image)

# 指定输入文件夹和输出文件夹的路径
input_folder = "./mask/"
output_folder = "./label/"

# 调用函数进行转换
convert_to_binary(input_folder, output_folder)