批量重命名图片/xml文件

简介

前段日子在用LabelImg对数据集完成标注后，我需要将两个文件的数据集图片放置在一个文件夹中，但是之前保存的两个文件中的图片存在相同文件名，那么就需要对文件名进行批量重命名，重命名后将文件放置在一个文件夹中。
更改前：

代码实现

完成了对图片及xml文件同时修改，修改后的命名符合COCO数据集的图片命名格式

import os


class BatchRename:

    def __init__(self):
    	// 图片及xml文件路径
        self.path = './Datasets/Pictures/Pictures/Defect'

    def rename(self):
        file_list = os.listdir(self.path)
        //没有file_list.sort()可能会导致重命名后的图片编号与之前不一致
        file_list.sort()
        total_num = len(file_list)
        i = 0
        for item in file_list:
            if item.endswith('.bmp'):
                src = os.path.join(os.path.abspath(self.path), item)
                if i < 10:
                    s = '00000000000' + str(i)
                elif 10 <= i < 100:
                    s = '0000000000' + str(i)
                else:
                    s = '000000000' + str(i)
                dst = os.path.join(os.path.abspath(self.path), s + '.bmp')

                try:
                    os.rename(src, dst)
                    print('converting %s to %s ...' % (src, dst))
                except:
                    continue
            if item.endswith('.xml'):
                src = os.path.join(os.path.abspath(self.path), item)
                if i < 10:
                    s = '00000000000' + str(i)
                elif 10 <= i < 100:
                    s = '0000000000' + str(i)
                else:
                    s = '000000000' + str(i)
                dst = os.path.join(os.path.abspath(self.path), s + '.xml')

                try:
                    os.rename(src, dst)
                    print('converting %s to %s ...' % (src, dst))
                    i = i + 1
                except:
                    continue
        print('total %d to rename & converted %d bmp' % (total_num, i))
        print('total %d to rename & converted %d xml' % (total_num, i))


if __name__ == '__main__':
    demo = BatchRename()
    demo.rename()

更改完后：

参考文章

https://blog.csdn.net/weixin_44354586/article/details/103784884