python批量修改文件名字

import os


def rename(path):

    # 图片编号从多少开始，不要跟VOC原本的编号重复了
    i = 1

    for filename in os.listdir(path):
        print(filename)

        if filename.endswith('.jpg') or filename.endswith('.bmp'):

            n = 4 - len(str(i))  #编号前面(n-1)个零

            img_name = path + filename
            img_rename = path + str(0) * n + str(i) + '.bmp' #改成需要的后缀

            # 重命名
            os.rename(img_name, img_rename)

            i += 1


        # break
    print('========================================================')
    # 检查更新后的名字
    for filename in os.listdir(path):
        print(filename)


if __name__ == '__main__':

   rename('F:\BS_IMAGE\demo/')

voc数据集的名字处理