Spaces:
Runtime error
Runtime error
import os | |
from shutil import copyfile | |
#copyfile(src, dst) | |
### list based on https://github.com/senguptaumd/Background-Matting/blob/master/Data_adobe/train_data_list.txt | |
train_list = open('train_data_list.txt').read().splitlines() | |
### training | |
src_fg_path = '/export/ccvl12b/qihang/MGMatting/data/Combined_Dataset/Training_set/fg/' | |
src_alpha_path = '/export/ccvl12b/qihang/MGMatting/data/Combined_Dataset/Training_set/alpha/' | |
dst_fg_path = '/export/ccvl12b/qihang/MGMatting/data/Combined_Dataset_Solid/Training_set/fg/' | |
dst_alpha_path = '/export/ccvl12b/qihang/MGMatting/data/Combined_Dataset_Solid/Training_set/alpha/' | |
if not os.path.exists(dst_fg_path): | |
os.makedirs(dst_fg_path) | |
if not os.path.exists(dst_alpha_path): | |
os.makedirs(dst_alpha_path) | |
for f in train_list: | |
copyfile(src_fg_path + f, dst_fg_path + f) | |
copyfile(src_alpha_path + f, dst_alpha_path + f) | |