【脚本】根据rootdir（或者一个列表），从文件夹1复制图像到文件夹2

# 根据rootdir，从文件夹1复制图像到文件夹2
import os
import shutil


def copyfile(rootdir, dir1, dir2):
    if isinstance(rootdir, list):  # 如果是直接传入的一个列表
        filelist = rootdir
    else:
        assert os.path.exists(rootdir), f'{rootdir} not exists.'
        filelist = os.listdir(rootdir)

    # ---------make dir2--------- #
    if not os.path.exists(dir2):
        os.makedirs(dir2)

    # ---------get dir1 ext------ #
    dir2_list = os.listdir(dir1)
    ext = os.path.splitext(dir2_list[0])[-1]

    cnt = 0
    for file in filelist:
        filename = os.path.splitext(file)[0]
        name = filename + ext
        shutil.copy(os.path.join(dir1, name), os.path.join(dir2, name))
        cnt += 1
        print(f'{file} has been copyed from {dir1} to {dir2}')
    print(f'copy num: {cnt}')


if __name__ == '__main__':
    rootdir = r'D:\A_dataset\liao_Upan\Concrete Crack Images for Classification\crack_json'
    dir1 = r'D:\A_dataset\liao_Upan\Concrete Crack Images for Classification\Positive'
    dir2 = r'D:\A_dataset\liao_Upan\Concrete Crack Images for Classification\Positive_96'
    copyfile(rootdir, dir1, dir2)