SketchyCOCO数据集进行前景图像、背景图像和全景图像的分类

import os
import shutildef CopyFile(src, dst, filename):if not os.path.exists(dst):os.makedirs(dst)print('create dir:' + dst)try:shutil.copy(src+'\\'+filename, dst+'\\'+filename)except Exception as e:print('copy fail:' + e)if __name__ == '__main__':current_path = os.getcwd()source_path = r"G:\SketchDiffusion\code\autodl-tmp\source"filename_list = os.listdir(source_path)for filename in filename_list:try:name1, name2 = filename.split('.')# foregroundif name2 == 'png':foreground_path = os.path.join(current_path, 'foreground')CopyFile(source_path, foreground_path, filename)# Panoramaelif name2 == 'jpg' and name1[:6] == "000000":panorama_path = os.path.join(current_path, 'panorama')CopyFile(source_path, panorama_path, filename)# backgroundelse:background_path = os.path.join(current_path, 'background')CopyFile(source_path, background_path, filename)except:passprint('over！')