时间:2020-07-30 python教程 查看: 1054
在做深度学习相关项目时,需要标注图片,筛选过后图片名字带有括号,显得比较乱,因此利用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博客。