File size: 1,491 Bytes
404d2af 8b973ee 404d2af 8b973ee 404d2af 8b973ee 404d2af 8b973ee 404d2af 8b973ee 404d2af 8b973ee 404d2af 8b973ee 404d2af |
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 43 44 45 46 47 |
"""
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)
|