Python之遍历文件夹图片并重命名

参考:https://blog.csdn.net/zhyh1435589631/article/details/51788350

在做深度学习相关项目时,需要标注图片,筛选过后图片名字带有括号,显得比较乱,因此利用python进行统一规范重命名操作


实现方法是利用python的os模块对文件夹进行遍历(listdir),然后使用rename进行改名操作

代码如下

# -*- coding:utf8 -*-
 
import os
 
class BatchRename():
    '''
    批量重命名文件夹中的图片文件
    '''
    def __init__(self):
        self.path = 'C:/Users/lenovo/Desktop/label5/3'
 
    def rename(self):
        filelist = os.listdir(self.path)
        total_num = len(filelist)
        i = 1
        for item in filelist:
            if item.endswith('.jpg'):
                src = os.path.join(os.path.abspath(self.path), item)
                dst = os.path.join(os.path.abspath(self.path), '02'+ str(i) + '.jpg')
                try:
                    os.rename(src, dst)
                    print 'converting %s to %s ...' % (src, dst)
                    group = 0
                except:
                    continue
            if item.endswith('.xml'):
                src = os.path.join(os.path.abspath(self.path), item)
                dst = os.path.join(os.path.abspath(self.path), '02'+ str(i) + '.xml')
                try:
                    os.rename(src, dst)
                    print 'converting %s to %s ...' % (src, dst)
                    group = 1
                    
                except:
                    continue
            if group:
                i = i + 1
        print 'total %d to rename & converted %d jpgs' % (total_num, i)
 
if __name__ == '__main__':
    demo = BatchRename()
    demo.rename()

运行后效果如下


猜你喜欢

转载自blog.csdn.net/zong596568821xp/article/details/80881237