【数据增强】剔除两个文件夹中不同名字的文件

import os
import shutil
from PIL import Image

def check_different_file(src, dst):
    for ori_name in os.listdir(src):
        is_exists = False
        for mask_name in os.listdir(dst):
            if ori_name.split('.')[0] == mask_name.split('.')[0]:
                is_exists = True
        if not is_exists:
            print(ori_name + ' is remove! ')
            os.remove(os.path.join(src, ori_name))