pycaffe-utils/pycaffe_config.py at master · pulkitag/pycaffe-utils · GitHub

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
## @package pycaffe_config
#  Specify the configurations
#

import socket
from easydict import EasyDict as edict
from os import path as osp

cfg = edict()
cfg.HOSTNAME = socket.gethostname()
if cfg.HOSTNAME in ['anakin', 'vader', 'spock', 'poseidon']:
	cfg.IS_EC2 = False
	cfg.CAFFE_PATH = '/work4/pulkitag-code/pkgs/caffe-v2-3'
	#cfg.CAFFE_PATH = '/work4/pulkitag-code/pkgs/caffe-jeff-dec15'
	cfg.STREETVIEW_CODE_PATH = '/work4/pulkitag-code/code/projStreetView'
	cfg.STREETVIEW_DATA_MAIN = '/data0'
	cfg.STREETVIEW_DATA_READ_IM = cfg.STREETVIEW_DATA_MAIN
	#Billiards Path
	cfg.BILLIARDS_DATA_MAIN = '/data1'
	cfg.DATA0 = '/data0'
	cfg.DATA1 = '/data1'
	cfg.BILLIARDS_CODE_PATH = '/work4/pulkitag-code'
	#Caffe Model Path
	cfg.CAFFE_MODEL_PATH = '/data1/pulkitag/caffe_models/'
else:
	cfg.IS_EC2 = True
	if osp.exists('/home-2/pagrawal'):
		cfg.STREETVIEW_CODE_PATH = '/home-2/pagrawal/code/streetview'
		cfg.CAFFE_PATH = '/home-2/pagrawal/pkgs/caffe-v2-3'
	else:
		cfg.STREETVIEW_CODE_PATH = '/home/ubuntu/code/streetview'
		cfg.CAFFE_PATH = '/home/ubuntu/caffe-v2-3'

	if osp.exists('/data0'):
		cfg.STREETVIEW_DATA_MAIN    = '/data0'
		cfg.STREETVIEW_DATA_READ_IM = cfg.STREETVIEW_DATA_MAIN
		#BILLIARDS PATH
		cfg.BILLIARDS_DATA_MAIN = '/data1'
		cfg.BILLIARDS_CODE_PATH = '/work4/pulkitag-code'
		cfg.DATA0 = '/data0'
		cfg.DATA1 = '/data1'
	else:
		cfg.STREETVIEW_DATA_MAIN    = '/puresan/shared'
		cfg.STREETVIEW_DATA_READ_IM = '/dev/shm'
		#BILLIARDA PATH
		cfg.BILLIARDS_DATA_MAIN = '/puresan/shared'
		cfg.BILLIARDS_CODE_PATH = '/home-2/pagrawal'
		cfg.DATA0 = '/dev/shm'
		cfg.DATA1 = '/dev/shm'
#Caffe Model Path
cfg.CAFFE_MODEL_PATH = osp.join(cfg.DATA0, 'pulkitag/caffe_models/')