diff options
author | Grimsley | 2018-08-01 08:44:34 -0500 |
---|---|---|
committer | Manu Mathew | 2019-12-13 06:41:03 -0600 |
commit | 8d00463d0be2857c831412633e241883049d458c (patch) | |
tree | 59dbe3b7df4cc0cb565d7ce2764b7e172874a29f | |
parent | 220d0dca7d0ba0a556c8a68202d8fe5a17b17b63 (diff) | |
download | caffe-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.py | 14 |
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 |