因为做模型训练时要对数据集进行处理,所以需要对图像文件进行重命名操作
可以选择覆盖或者不覆盖原文件
代码如下:
import os
import shutil
class BatchRename():
'''
批量重命名文件夹中的图片文件
'''
def __init__(self):
self.path = 'G:/sar2optical/test' # 表示需要命名处理的文件夹
self.save_path = 'G:/sar2optical/output' # 保存重命名后的图片地址
def rename(self):
filelist = os.listdir(self.path) # 获取文件路径
total_num = len(filelist) # 获取文件长度(个数)
i = 1 # 表示文件是从1开始的
for item in filelist:
print(item)
if item.endswith('.png'): # 初始的图片的格式为jpg格式的(或者源文件是png格式及其他格式,后面的转换格式就可以调整为自己需要的格式即可)
src = os.path.join(os.path.abspath(self.path), item) # 当前文件中图片的地址
dst = os.path.join(os.path.abspath(self.save_path), item.split("p")[1] + 'png') # 处理后文件的地址和名称,可以自己按照自己的要求改进
try:
# os.rename(src, dst) 覆盖原文件
shutil.copy(src, dst) #不覆盖原文件
i = i + 1
except:
continue
print('运行完成!共 %d 张图片 重命名为 %d 张图片' % (total_num, i-1))
if __name__ == '__main__':
demo = BatchRename()
demo.rename()