批量颜色取反--把白色换成黑色，黑色换成白色

原图：

效果图：

代码：

'''2020.9.15
把白色背景换成黑色，把其他黑色的细胞换成白色标记
'''
import cv2
import os

data_dir = './Fold_1/masks/fold1/channel_5/'#源文件路径
save_path = './Fold_1/masks/fold1/change_background/'#文件保存路径

img_ids = sorted(os.listdir(data_dir))#给文件排序，方便取出名字
print(len(img_ids))#打印出文件夹下的文件个数

for img_id in img_ids:
    dataFile = data_dir + img_id  # 单个的文件
    img = cv2.imread(dataFile,1)# 以彩色图读入
    for row in range(img.shape[0]):
        for col in range(img.shape[1]):
            for c in range(img.shape[2]):
                pv = img[row, col, c]# 分别取出RGB三个通道
                img[row, col, c] = 255 - pv  # 取反 白的变成黑的，黑的变成白的
    cv2.imwrite(save_path+str(img_id),img)#保存

    if cv2.waitKey(0)&0xFF ==27:#按Esc键退出
        cv2.destroyAllWindows()