用Python将图形中的红色和绿色曲线去除

这是测量数据的原始图片

 这是处理之后的图片

可以通过程序对文件夹内所有的图片进行批处理

 

用进度条显示处理进度

以下是程序代码

# 导入所需的库
import os
import cv2
from PIL import Image
import numpy as np
from tqdm import tqdm

# 设置包含图像的目录路径和输出目录路径
directory = "替换为图像保存目录"
directory_modified = "替换为图像保存目录"

# 获取目录中所有PNG图像文件的列表
image_files = [os.path.join(directory, f) for f in os.listdir(directory) if f.endswith(".png")]

# 定义要替换的颜色及其容差
red = (255, 0, 0)
red_tolerance = [(red[0]-120, red[1]-120, red[2]-120), (red[0]+120, red[1]+120, red[2]+120)]
green = (0, 255, 0)
green_tolerance = [(green[0]-120, green[1]-120, green[2]-120), (green[0]+120, green[1]+100, green[2]+120)]
blue = (0, 0, 255)
blue_tolerance = [(blue[0]-150, blue[1]-150, blue[2]-150), (blue[0]+150, blue[1]+150, blue[2]+150)]

# 遍历每个图像文件
tqdm_kwargs = {
    "bar_format": "{l_bar}{bar}{r_bar}",
    "desc": "Processing images",
    "colour": "#00ff00"
}
for image_file in tqdm(image_files, **tqdm_kwargs):