Spaces:
Running
Running
File size: 1,480 Bytes
a80d6bb |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 |
"""
Project configurations.
"""
import os
class Config(object):
""" Datasets and experiments folders for the whole project. """
#####################
## Dataset setting ##
#####################
DATASET_ROOT = os.getenv("DATASET_ROOT", "./datasets/") # TODO: path to your datasets folder
if not os.path.exists(DATASET_ROOT):
os.makedirs(DATASET_ROOT)
# Synthetic shape dataset
synthetic_dataroot = os.path.join(DATASET_ROOT, "synthetic_shapes")
synthetic_cache_path = os.path.join(DATASET_ROOT, "synthetic_shapes")
if not os.path.exists(synthetic_dataroot):
os.makedirs(synthetic_dataroot)
# Exported predictions dataset
export_dataroot = os.path.join(DATASET_ROOT, "export_datasets")
export_cache_path = os.path.join(DATASET_ROOT, "export_datasets")
if not os.path.exists(export_dataroot):
os.makedirs(export_dataroot)
# Wireframe dataset
wireframe_dataroot = os.path.join(DATASET_ROOT, "wireframe")
wireframe_cache_path = os.path.join(DATASET_ROOT, "wireframe")
# Holicity dataset
holicity_dataroot = os.path.join(DATASET_ROOT, "Holicity")
holicity_cache_path = os.path.join(DATASET_ROOT, "Holicity")
########################
## Experiment Setting ##
########################
EXP_PATH = os.getenv("EXP_PATH", "./experiments/") # TODO: path to your experiments folder
if not os.path.exists(EXP_PATH):
os.makedirs(EXP_PATH)
|