一、需求

深度学习配对训练，有时配对图像的名称需要一致，这里写了一个脚本，快速批量替换图片名称中某些字符串。

二、代码使用方法

使用代码时需要修改的地方见下：

三、代码

完整代码见下：

import os
import cv2
import shutil

folder_path = '/media/Work/Dataset/'
output_path = '/media/Work/Dataset/rename/'

# 遍历文件夹中所有图像
for filename in os.listdir(folder_path):
    # 如果文件名包含CLAHE
    if "ResizeNEAREST_GaussianBlur" in filename:
        # 构建新的文件名
        # new_filename = filename.replace('CLAHE','OneDegradation')
        new_filename = filename.replace('ResizeNEAREST_GaussianBlur','USMr80w0.9t30')

        sortce_path = os.path.join(folder_path,filename)
        img = cv2.imread(sortce_path,-1)


        print(new_filename)
        cv2.imwrite(os.path.join(output_path,new_filename),img)

四、效果展示