-
Notifications
You must be signed in to change notification settings - Fork 0
/
config.py
66 lines (47 loc) · 2.2 KB
/
config.py
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
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
#!/usr/bin/env python
# Core libraries
import os
from easydict import EasyDict as edict
"""
This file is for managing ALL constants and configuration values
They're all nested per python class
To import the file:
from config import cfg
easydict usage:
https://pypi.python.org/pypi/easydict/
"""
# The main configuration variable / dictionary
cfg = edict()
##########################################################################################
##########################################################################################
##########################################################################################
##########################################################################################
"""
Generic constants & parameters
"""
cfg.GEN = edict()
# User name on Blue Pebble/Crystal (CHANGE THIS TO YOUR OWN USERNAME)
cfg.GEN.BLUE_USERNAME = "fo18103"
"""
Dataset constants
"""
# The base dictionary
cfg.DATASET = edict()
# # Where to find the RGBDCows2020 dataset
# if os.path.exists("/home/io18230/Desktop"): # Home Windows machine
# cfg.DATASET.RGBDCOWS2020_LOC = "/home/io18230/Desktop/RGBDCows2020/ideal"
# elif os.path.exists(f"/work/{cfg.GEN.BLUE_USERNAME}/Projects/ATI-workspace/datasets/RGBDCows2020/ideal"): # Blue Pebble
# cfg.DATASET.RGBDCOWS2020_LOC = f"/work/{cfg.GEN.BLUE_USERNAME}/Projects/ATI-workspace/datasets/RGBDCows2020/ideal"
#
#
# # Where to find the validation dataset
# if os.path.exists("/home/io18230/Desktop"): # Home Windows machine
# cfg.DATASET.RGBDCOWS2020_val2 = "/home/io18230/Desktop/RGBDCows2020_val23/Identification"
# elif os.path.exists(f"/work/{cfg.GEN.BLUE_USERNAME}/Projects/ATI-workspace/datasets/RGBDCows2020_val23/Identification"): # Blue Pebble
# cfg.DATASET.RGBDCOWS2020_val2 = f"/work/{cfg.GEN.BLUE_USERNAME}/Projects/ATI-workspace/datasets/RGBDCows2020_val23/Identification"
#
#
# Where to find the RGBDCows2020 dataset
cfg.DATASET.RGBDCOWS2020_LOC = f"/user/work/{cfg.GEN.BLUE_USERNAME}/cows2021_dataset/Sub-levels/Identification/Train/RGBDCows2020/Identification"
# Where to find the validation dataset
cfg.DATASET.RGBDCOWS2020_val2 = f"/user/work/{cfg.GEN.BLUE_USERNAME}/cows2021_dataset/Sub-levels/Identification/Train/RGBDCows2020/Identification"