-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathconstants.py
26 lines (20 loc) · 1.05 KB
/
constants.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
# Work space directory
HOME_DIR = '/home/siri/PycharmProjects/ECSSGAN/'
# Path to SALICON raw data
pathToImages = '/home/siri/PycharmProjects/ECSSGAN/scripts/images/'
pathToMaps = '//home/siri/PycharmProjects/ECSSGAN/ground_truth_mask/'
# pathToFixationMaps = '/home/users/jpang/salicon_data/fixation'
# Path to processed data
pathOutputImages = '/home/siri/PycharmProjects/ECSSGAN/output/'
pathOutputMaps = '/home/siri/PycharmProjects/ECSSGAN/maps/'
pathToPickle = '/home/siri/PycharmProjects/ECSSGAN/pickle/'
# Path to pickles which contains processed data
TRAIN_DATA_DIR = '/home/siri/PycharmProjects/ECSSGAN/data_test_directory/fix_trainData.pickle'
VAL_DATA_DIR = '/home/siri/PycharmProjects/ECSSGAN/data_test_directory/fix_validationData.pickle'
TEST_DATA_DIR = '/home/siri/PycharmProjects/ECSSGAN/data_test_directory/testData.pickle'
# Path to vgg16 pre-trained weights
PATH_TO_VGG16_WEIGHTS = '/home/siri/PycharmProjects/ECSSGAN/scripts/vgg16.pkl'
# Input image and saliency map size
INPUT_SIZE = (256, 192)
# Directory to keep snapshots
DIR_TO_SAVE = 'test'