深度学习批量处理重命名图片（xml，jpg）

yolo3图片数据集处理

 

视频拆解成帧图片时，文件重命名有时候会比较头疼。

通过代码批处理文件名称

如果是jpg相应修改xml-jpg

import os

import random


class ImageRename():
    def __init__(self):

        self.path = 'D:\shipincunfang\shujuji'  # 图片所在文件夹

    def rename(self):

        filelist = os.listdir(self.path)

        random.shuffle(filelist)

        total_num = len(filelist)

        i = 0

        for item in filelist:

            if item.endswith('.xml'):
                src = os.path.join(os.path.abspath(self.path), item)

                dst = os.path.join(os.path.abspath(self.path), '0000' + format(str(i), '0>3s') + '.xml')

                os.rename(src, dst)

                print('converting %s to %s ...' % (src, dst))

                i = i + 1

        print('total %d to rename & converted %d xml' % (total_num, i))


if __name__ == '__main__':
    newname = ImageRename()

    newname.rename()