批量将图片转为黑白图

'''
可批量将图片转黑白
'''
import os
import shutil
import cv2
# 输入你的文件路径
label_from_PATH = ''
label_to_PATH = ' '

filepath_list = os.listdir(label_from_PATH)
# filepath_list.remove("labelme_json_to_dataset.exe")

# 检查是否存在label文件夹
if not os.path.isdir(label_to_PATH):
    os.mkdir(label_to_PATH)

# 是否图像二值化
bin_img = True

for i, file_path in enumerate(filepath_list):
    src_label = "{}".format(os.path.join(label_from_PATH, filepath_list[i]))
    label_name = "{}.png".format(file_path)
    if bin_img:
        dest_label = cv2.imread(src_label)
        dest_label = cv2.cvtColor(dest_label, cv2.COLOR_BGR2GRAY)
        ret, dest_label = cv2.threshold(dest_label, 0, 255, cv2.THRESH_BINARY)
        cv2.imwrite(os.path.join(label_to_PATH, label_name), dest_label)
    else:
        shutil.copy(src_label, os.path.join(label_to_PATH, label_name))

    print("{} has been copied to {}".format(label_name, label_to_PATH))

print("All done!!!")