批量修改文件内图片名

import os

#批量修改文件名字


s = 'PreviewData'
r='RGB'
#for line in open("/media/qq/work/flir/data/ImageSets/train.txt", "r"):
    #line
    # 'FLIR_09593_PreviewData
    #_RGB
    #if s in fileName:

def add_prefix_to_images(directory, prefix):
    for filename in os.listdir(directory):
        if filename.endswith(('.jpg','.txt','.jpeg', '.png', '.gif', '.bmp', '.tif', '.tiff')):
            filename_change = filename.replace(r, s) #修改之后的文件按名字
            os.rename(os.path.join(directory, filename), os.path.join(directory, filename_change))


# 使用示例
directory = '/media/qq/work/flir/data/infrared/test/'  # 图片所在目录
prefix = '2'  # 要添加的前缀
add_prefix_to_images(directory, prefix)