文件夹中图像重命名

import os.path

def rename(img_folder,num):
    for img_name in os.listdir(img_folder):  # os.listdir()： 列出路径下所有的文件
        #os.path.join() 拼接文件路径
        #src：原文件名
        #dst：新文件名
        src = os.path.join(img_folder, img_name)
        dst = os.path.join(img_folder,str(num) + '.jpg')
        print(num)
        num= num+1
        os.rename(src, dst) #用dst替代src
        


def main():
    #图片的文件夹路径
    img_folder0 = '/home/ly/CLTR_OUR/data/dataset/test/images'
    num=1
    rename(img_folder0,num)

if __name__=="__main__":
    main()