当前位置:首页 >> 脚本专栏

python实现遍历文件夹图片并重命名

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

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()

运行后效果如下

python实现遍历文件夹图片并重命名

简洁版(tab补全)

# -*- coding:utf8 -*-
import os
path = 'SegmentationClassRaw/'
filelist = os.listdir(path)
for item in filelist:
 #print('item name is ',item)
 if item.endswith('.png'):
 name = item.split('.',3)[0] + '.' + item.split('.',3)[1]
 src = os.path.join(os.path.abspath(path),item)
 dst = os.path.join(os.path.abspath(path),name + '.png')
 try:
 os.rename(src,dst)
 print('rename from %s to %s'%(src,dst))
 except:
 continue

参考:利用Python对文件夹下图片数据进行批量改名的代码实例

本文已被收录到专题《python图片处理操作》 ,欢迎大家点击学习更多精彩内容。

以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持。