批量处理voc数据imagest中图片的名字

-- coding:utf8 --

import os

class BatchRename():
‘’’
批量重命名文件夹中的图片文件
‘’’

def __init__(self):
    # 我的图片文件夹路径horse
    self.path = 'C:\\Users\\RENCONG\\Desktop\\VOCdevkit\\VOC2007\\JPEGImages'

def rename(self):
    filelist = os.listdir(self.path)
    total_num = len(filelist)
    i = 1
    n = 6
    for item in filelist:
        if item.endswith('.jpg'):
            n = 6 - len(str(i))
            src = os.path.join(os.path.abspath(self.path), item)
            dst = os.path.join(os.path.abspath(self.path), str(0) * n + str(i) + '.jpg')
            try:
                os.rename(src, dst)
                print
                'converting %s to %s ...' % (src, dst)
                i = i + 1

            except:
                continue
    print
    'total %d to rename & converted %d jpgs' % (total_num, i)

if name == ‘main’:
demo = BatchRename()
demo.rename()