def img_mark(img_path):
img_data = Image.open(img_path)
width, height = img_data.size # 获取图片宽高
for p in product(range(width), range(height)):
dian_data = img_data.getpixel(p)[:3]
if 50 < dian_data[0] <= 200 and 60 < dian_data[1] < 105 and 0 < dian_data[2] < 80: # 进行阈值筛选
img_data.putpixel(p, (127, 121, 107)) # 如果满足替换为底色
img_data.save('_OK'.join(splitext(img_path)))
阈值需要自己查找,可以手动,也可以使用经过训练后的值
如果水印颜色较多,可以多加几个判断