aboutsummaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
authorGrimsley2018-08-01 08:44:34 -0500
committerManu Mathew2019-12-13 06:41:03 -0600
commit8d00463d0be2857c831412633e241883049d458c (patch)
tree59dbe3b7df4cc0cb565d7ce2764b7e172874a29f
parent220d0dca7d0ba0a556c8a68202d8fe5a17b17b63 (diff)
downloadcaffe-jacinto-models-8d00463d0be2857c831412633e241883049d458c.tar.gz
caffe-jacinto-models-8d00463d0be2857c831412633e241883049d458c.tar.xz
caffe-jacinto-models-8d00463d0be2857c831412633e241883049d458c.zip
fix parameter override by default
-rw-r--r--scripts/models/image_segmentation.py14
1 files changed, 7 insertions, 7 deletions
diff --git a/scripts/models/image_segmentation.py b/scripts/models/image_segmentation.py
index f1ec9d2..fff95cd 100644
--- a/scripts/models/image_segmentation.py
+++ b/scripts/models/image_segmentation.py
@@ -98,7 +98,12 @@ def main():
98 98
99 #more threads doesn't seem to help speed up - but you can try increasing this, at the cost of system becoming more slugginsh to respond if you are doing something else parallely. 99 #more threads doesn't seem to help speed up - but you can try increasing this, at the cost of system becoming more slugginsh to respond if you are doing something else parallely.
100 config_param.threads = 1 #4 100 config_param.threads = 1 #4
101 101
102 config_param.train_data = "data/train-image-list.txt" if config_param.use_image_list else 'data/train-image-lmdb'
103 config_param.train_label = "data/train-label-list.txt" if config_param.use_image_list else 'data/train-label-lmdb'
104 config_param.test_data = "data/val-image-list.txt" if config_param.use_image_list else 'data/val-image-lmdb'
105 config_param.test_label = "data/val-label-list.txt" if config_param.use_image_list else 'data/val-label-lmdb'
106
102 #Update from params given from outside 107 #Update from params given from outside
103 #if args.config_param != None: 108 #if args.config_param != None:
104 # config_param.update(args.config_param) 109 # config_param.update(args.config_param)
@@ -106,12 +111,7 @@ def main():
106 for k in args.config_param.keys(): 111 for k in args.config_param.keys():
107 config_param.__setattr__(k,args.config_param[k]) 112 config_param.__setattr__(k,args.config_param[k])
108 config_param.__setitem__(k,args.config_param[k]) 113 config_param.__setitem__(k,args.config_param[k])
109 114
110 config_param.train_data = "data/train-image-list.txt" if config_param.use_image_list else 'data/train-image-lmdb'
111 config_param.train_label = "data/train-label-list.txt" if config_param.use_image_list else 'data/train-label-lmdb'
112 config_param.test_data = "data/val-image-list.txt" if config_param.use_image_list else 'data/val-image-lmdb'
113 config_param.test_label = "data/val-label-list.txt" if config_param.use_image_list else 'data/val-label-lmdb'
114
115 # Modify the job name if you want. 115 # Modify the job name if you want.
116 config_param.base_name = config_param.config_name 116 config_param.base_name = config_param.config_name
117 config_param.job_name = config_param.base_name 117 config_param.job_name = config_param.base_name