correction

Maxlo24 · Jul 14, 2022 · 2c66a08 · 2c66a08
1 parent fa4b370
commit 2c66a08
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/src/AGENT/GlobalVar.py b/src/AGENT/GlobalVar.py
@@ -30,7 +30,7 @@
 # LABELS_TO_TRAIN = new_LABELS
 
 
-
+LABELS_TO_TRAIN = ['RAF','RAE']
 
 LABEL_GROUPES = {}
 LABELS = []

diff --git a/src/AGENT/predic_landmarks.py → src/AGENT/predict_landmarks.py b/src/AGENT/predic_landmarks.py → src/AGENT/predict_landmarks.py
@@ -133,7 +133,7 @@ def main(args):
 
 
     # agent_lst = GetAgentLst(agents_param)
-    brain_lst = GetBrain(args.dir_model)
+    brain_lst = GetBrain(args.dir_models)
     # print( brain_lst)
     # environement_lst = [environement_lst[0]]
     # agent_lst = [agent_lst[0]]
@@ -200,9 +200,9 @@ def main(args):
 
     input_group = parser.add_argument_group('dir')
     input_group.add_argument('--dir_scans', type=str, help='Input directory with the scans',default='/Users/luciacev-admin/Documents/Projects/MSDRL_benchmark/data/test')
-    input_group.add_argument('--dir_model', type=str, help='Directory of the trained models',default= '/Users/luciacev-admin/Desktop/MSDRL_models/ALI_CNN_models_2021_26_10')
+    input_group.add_argument('--dir_models', type=str, help='Directory of the trained models',default= '/Users/luciacev-admin/Desktop/MSDRL_models/ALI_CNN_models_2021_26_10')
 
-    input_group.add_argument('--clear_temp', type=bool, help='Temp directory',default= False)
+    input_group.add_argument('--clear_temp', type=bool, help='Temp directory',default= True)
     input_group.add_argument('--dir_temp', type=str, help='Temp directory',default= '..')