def source_and_add_filepath_define(source_dir_path, add_dir_path):
source_dir = os.listdir(source_dir_path) # 读取source路径下的全部文件
add_dir = os.listdir(add_dir_path) # 读取add文件夹下的全部文件
# print(path)
isafterfolderexist = os.path.exists(
add_dir_path + "/after") # add文件夹路径下是否存在after文件夹，如果不存在则创建
if not isafterfolderexist:
os.makedirs(add_dir_path + "/after")
with open(add_dir_path + r"\rgb.json", 'r') as load_f:
load_dict = json.load(load_f)
rgb = load_dict["rgb"]
rgb = [rgb[2], rgb[1], rgb[0]]
for imgname in source_dir:
if (imgname == "add" or imgname == "after" or imgname == "rbg.json"):
continue
print(imgname)
shortName = imgname.split(".")[0]
add_image_name = shortName + "_add.bmp"
if add_image_name not in add_dir:
add_image_name = shortName + "_add.BMP"
if add_image_name not in add_dir:
messagebox.showinfo("错误", shortName + "不存在对应的add文件")
continue
mask_change(source_dir_path + "/" + imgname,
add_dir_path + "/" + add_image_name,
add_dir_path + "/after/" + shortName + ".bmp", rgb)
def add_file_define(dir_path):
dir = os.listdir(dir_path) # 读取总路径下的全部文件
source_dir_path = dir_path + r"/source" # 最开始的source文件
for i in range(1, 10): # 默认最多有10个add类
add_dir_path = dir_path + r"/add_" + str(i)
print(add_dir_path)
if (r"add_" + str(i) not in dir):
messagebox.showinfo("提示", r"操作完毕")
break
if i != 1:
# 每轮操作后，都选择新的source文件夹和新的add文件夹
source_dir_path = dir_path + r"/add_" + str(i - 1) + r"/after"
source_and_add_filepath_define(source_dir_path, add_dir_path)
file.write(str(add_mask_tem[i][j]) + " ")
file.write("\n")
# numpy.savetxt("result", add_mask_tem)
# len(add_mask_tem)
# for i in range(len(add_mask_tem)):
# file.write(str(add_mask_tem[i]))
file.close()
def getFilePath(add_dir_path):
add_dir = os.listdir(add_dir_path)
with open(add_dir_path + r"\rgb.json", 'r') as load_f:
load_dict = json.load(load_f)
rgb = load_dict["rgb"]
rgb = [rgb[2], rgb[1], rgb[0]]
for add_image_name in add_dir:
if "bmp" not in add_image_name:
continue
getMatrix(add_dir_path + "/" + add_image_name, rgb)
messagebox.showinfo("提示", r"请选择文件夹")
# root = tk.Tk()
# root.withdraw()
file_path = filedialog.askdirectory()
getFilePath(file_path)
def mask_change(source_img_path,add_img_path,save_path,rgb):
source_image = cv2.imread(source_img_path)
add_image = cv2.imread(add_img_path)
# 将界限值往左右移动10单位
color_lower = [i-10 for i in rgb]
print(color_lower)
color_upper = [i+10 for i in rgb]
print(color_upper)
lower_blue = np.array(color_lower)
upper_blue = np.array(color_upper)
# 将add所选区域涂白,其他地方都是黑的
add_mask_tem = cv2.inRange(add_image, lower_blue, upper_blue)
# 求轮廓
add_img_area, add_contours_area, add_hierarchy_area = cv2.findContours(add_mask_tem, cv2.RETR_TREE, cv2.CHAIN_APPROX_NONE)
# 求面积
add_allarea = cv2.contourArea(add_contours_area[0])
# 将source图片里面的黑色部分全部涂白，其他地方为黑色,奖source文件里面的（0,0,0）-（30,30,30）范围视为黑色
source_mask_tem = cv2.inRange(source_image, np.array([0, 0, 0]), np.array([30, 30, 30]))
source_mask = 255 - source_mask_tem
##########
# cv2.waitKey(0)
# 膨胀操作的卷集核
# kernel = cv2.getStructuringElement(cv2.MORPH_RECT, (5,5))
# 膨胀操作
# mask_binary_dilation = cv2.dilate(mask, kernel)
#
# add_image_dilation = cv2.dilate(add_image, kernel)
#
# img_mask_rgb = cv2.bitwise_and(add_image_dilation, add_image_dilation, mask=mask_binary_dilation)
##########
# 将黑白化的add图片和source图片取交集（不允许add添加区域超过source中已经涂色的区域），取出真正需要的部分为白色，其余地方为黑色
add_mask = cv2.bitwise_and(source_mask_tem, add_mask_tem)
# 得到超出的部分，也就是涂色越界的部分
add_mask_xor = cv2.bitwise_xor(add_mask, add_mask_tem)
xor_img, xor_contours, xor_hierarchy = cv2.findContours(add_mask_xor, cv2.RETR_TREE, cv2.CHAIN_APPROX_NONE)
xor_area = 0;
if len(xor_contours) is 0:
xor_area = 0
else:
for i in range(len(xor_contours)):
xor_area += cv2.contourArea(xor_contours[i])
# 越界的部分的面积占新增全部部分的面积的比例
if(xor_area/add_allarea > 0.01):
messagebox.showinfo("错误", add_img_path+"\n 新增面积对原有有色区域有大面积覆盖,请检查你的涂色结果")
return
# 取出add图片中真正有效的涂色区域
add_mask_effective = cv2.bitwise_and(add_image, add_image, mask=add_mask)
source_image_mask = cv2.bitwise_and(source_image, source_image,mask=source_mask)
# 有效涂色区域和source文件叠加
final_image = add_mask_effective + source_image_mask
cv2.imwrite(save_path, final_image)