-
Notifications
You must be signed in to change notification settings - Fork 17
Expand file tree
/
Copy pathgenerate_dataset_csv.py
More file actions
19 lines (17 loc) · 1002 Bytes
/
Copy pathgenerate_dataset_csv.py
File metadata and controls
19 lines (17 loc) · 1002 Bytes
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
from data_loading import generate_sfsnet_data_csv, generate_celeba_data_csv, generate_celeba_synthesize_data_csv
'''
# SfSNet dataset
dataset_path = '/nfs/bigdisk/bsonawane/sfsnet_data/'
# dataset_path = '../data/sfs-net/'
generate_sfsnet_data_csv(dataset_path + 'train/', dataset_path + '/train.csv')
generate_sfsnet_data_csv(dataset_path + 'test/', dataset_path + '/test.csv')
# CelebA dataset
dataset_path = '/nfs/bigdisk/bsonawane/CelebA-dataset/CelebA_crop_resize_128/'
# dataset_path = '../data/celeba/'
generate_celeba_data_csv(dataset_path + 'train/', dataset_path + '/train.csv')
generate_celeba_data_csv(dataset_path + 'test/', dataset_path + '/test.csv')
'''
dataset_path = '/nfs/bigdisk/bsonawane/CelebA-dataset/CelebA_crop_resize_128/sfs_synthesized_data/'
# dataset_path = '../data/celeba/synthesized_data/'
generate_celeba_synthesize_data_csv(dataset_path + 'train/', dataset_path + '/train.csv')
generate_celeba_synthesize_data_csv(dataset_path + 'test/', dataset_path + '/test.csv')