File size: 379 Bytes
938e515 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 |
import os
DATASET = 'VIP' # DATASET: MHPv2 or CIHP or VIP
TYPE = 'crop_pic' # crop_pic or DemoDataset
IMG_DIR = '../demo/cropped_img/crop_pic'
SAVE_DIR = '../demo/cropped_img'
if not os.path.exists(SAVE_DIR):
os.makedirs(SAVE_DIR)
with open(os.path.join(SAVE_DIR, TYPE + '.txt'), "w") as f:
for img_name in os.listdir(IMG_DIR):
f.write(img_name[:-4] + '\n')
|