'''
功能:读取文件夹里的所有文件,判断是不是图片文件,然后进行图片反转,更改文件名后保存到指定文件夹中
入参:
file_compare:读取文件路径,
save_Path:保存文件路径
输出图片
''’
import os
import cv2
import numpy as np
def deal_Image( file_compare, save_Path):
for i in os.listdir(file_compare):
image_Path = os.path.join(file_compare, i)
string_name = os.path.splitext(i)
name = {'.jpg', '.bmp', '.png', '.jpeg', '.rgb', '.tif'}
if string_name[1] in name:
img = cv2.imread(image_Path, -1)
img = 255 - img
cv2.imwrite(save_Path + "/" + string_name[0]+"_label"+".png", img)
if __name__ == '__main__':
file_compare = r'D:\xuexi1\region'
save_Path = r'D:\xuexi1\model'
deal_Image(file_compare,save_Path)