python 如何把包含图片和txt两种格式的文件重新命名并保存-yolo格式

原图：
效果：

'''
批量重命名文件夹中的图片文件
'''

import os


class BatchRename():

    def __init__(self):
        self.path1 = r'E:\01_hjz\datas\facemask\mask10000_raw'  #原始文件夹路径，下面有图片和对应的txt，名称混乱

    # self.path2 = r'F:/fire_data/SSD_fire_data/firename/'

    def rename(self):
        filelist = os.listdir(self.path1)
        # print(filelist)
        total_num = len(filelist)
        print(total_num)
        i = 1
        j = 1
        #
        for item in filelist:
            if item.endswith('.txt'):
                src = os.path.join(os.path.abspath(self.path1), item)
                str1 = str(i)
                dst = os.path.join(os.path.abspath(self.path1), str1.zfill(6) + '.txt')  # str1.zfill(x),x为一共几位数，用0补齐，如001000

                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 item.endswith('.jpg'):
                src = os.path.join(os.path.abspath(self.path1), item)
                str1 = str(j)
                dst = os.path.join(os.path.abspath(self.path1), str1.zfill(6) + '.jpg')  # str1.zfill(x),x为一共几位数，用0补齐，如001000

                try:
                    os.rename(src, dst)
                    print('converting %s to %s ...' % (src, dst))
                    j = j + 1
                except:
                    continue


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

参考文章：
【python】批量重命名文件夹中的所有图片名称和文件夹中所有的txt文件
目标检测——口罩识别数据集