当前位置:首页 » python教程 » 正文

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

看: 1055次  时间:2020-07-30  分类 : 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()

运行后效果如下

简洁版(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博客。

<< 上一篇 下一篇 >>

搜索

推荐资源

  Powered By python教程网   鲁ICP备18013710号
python博客 - 小白学python最友好的网站!