python 文件夾圖片重命名

python 文件夾圖片重命名

import os


# 重命名 文件夾/000.jpg 模式
def rename(path):
    file_list = os.listdir(path)
    print(file_list)
    count = 0
    for item in file_list:
        src = os.path.join(os.path.abspath(path), item)
        dst = os.path.join(os.path.abspath(path), '%05d.jpg' % count)
        try:
            os.rename(src, dst)
            count += 1
        except:
            continue


# 重命名  根目錄/文件夾/000.jpg 模式
def rename_root(root):
    for dir_name in os.listdir(root):
        path_dir = root + dir_name + "/"
        print(path_dir)
        rename(path_dir)


def main():
    # 測試使用
    root = "img_train/"
    rename_root(root)

    
if __name__ == '__main__':
    main()