python将多张mask图片黑白颠倒 （cv2.bitwise_not）

# #!/usr/bin/python
# #coding=utf-8

import cv2
import os

img_dir = "XXX" # 输入mask图片所在文件夹路径
img_list = os.listdir(img_dir) # 获取所有图片名字
img_dir_out = "/home/cue/Documents/dlzr_NC/dlzr_4k_out_out/" # 输出mask图片所在文件夹路径

if __name__ == '__main__':
		# 依次读取图片
    for i in range(0, len(img_list)):
        img_name = img_list[i]
        img_path = os.path.join(img_dir,img_name)
    
        img = cv2.imread(img_path)  # 读入图片
        img_conv = cv2.bitwise_not(src=img)  # mask图片黑白颠倒

        cv2.imwrite(img_dir_out + img_name, img_conv)  # 输出图片