文件夹内图片批量重命名代码

文件夹内图片批量重命名代码

import os
import re
import sys
import cv2
import torchvision.transforms as transforms
path = r"data/masks/"
def renameall(path):
    fileList = os.listdir(path)  # 待修改文件夹
    currentpath = os.getcwd()  # 得到进程当前工作目录
    os.chdir(path)
    for fileName in fileList:  # 遍历文件夹中所有文件
        filenamenew = ''
        for i in range(0, len(fileName)):
            if fileName[i] !='_':
                filenamenew += fileName[i]
            else:
                break
        filenamenew += '.png'
        os.rename(fileName, filenamenew)  # 文件重新命名
    os.chdir(currentpath)  # 改回程序运行前的工作目录
    sys.stdin.flush()  # 刷新
renameall(path)