implementing model switch

app.py

@@ -44,32 +44,35 @@ def plot_bbox(bbox_XYXY, label):
 def atomlenz(modelfile):
     model_cls = RCNN
     experiment_path_atoms="./models/atoms_model/"
-    dir_list = os.listdir(experiment_path_atoms)
-    dir_list = [os.path.join(experiment_path_atoms,f) for f in dir_list]
-    dir_list.sort(key=os.path.getctime, reverse=True)
-    checkpoint_file_atoms = [f for f in dir_list if "ckpt" in f][0]
+    #dir_list = os.listdir(experiment_path_atoms)
+    #dir_list = [os.path.join(experiment_path_atoms,f) for f in dir_list]
+    #dir_list.sort(key=os.path.getctime, reverse=True)
+    #checkpoint_file_atoms = [f for f in dir_list if "ckpt" in f][0]
     checkpoint_file_atoms=os.path.join(experiment_path_atoms,modelfile)
     model_atom = model_cls.load_from_checkpoint(checkpoint_file_atoms)
     model_atom.model.roi_heads.score_thresh = 0.65
     experiment_path_bonds = "./models/bonds_model/"
-    dir_list = os.listdir(experiment_path_bonds)
-    dir_list = [os.path.join(experiment_path_bonds,f) for f in dir_list]
-    dir_list.sort(key=os.path.getctime, reverse=True)
-    checkpoint_file_bonds = [f for f in dir_list if "ckpt" in f][0]
+    #dir_list = os.listdir(experiment_path_bonds)
+    #dir_list = [os.path.join(experiment_path_bonds,f) for f in dir_list]
+    #dir_list.sort(key=os.path.getctime, reverse=True)
+    #checkpoint_file_bonds = [f for f in dir_list if "ckpt" in f][0]
+    checkpoint_file_bonds=os.path.join(experiment_path_bonds,modelfile)
     model_bond = model_cls.load_from_checkpoint(checkpoint_file_bonds)
     model_bond.model.roi_heads.score_thresh = 0.65
     experiment_path_stereo = "./models/stereos_model/"
-    dir_list = os.listdir(experiment_path_stereo)
-    dir_list = [os.path.join(experiment_path_stereo,f) for f in dir_list]
-    dir_list.sort(key=os.path.getctime, reverse=True)
-    checkpoint_file_stereo = [f for f in dir_list if "ckpt" in f][0]
+    #dir_list = os.listdir(experiment_path_stereo)
+    #dir_list = [os.path.join(experiment_path_stereo,f) for f in dir_list]
+    #dir_list.sort(key=os.path.getctime, reverse=True)
+    #checkpoint_file_stereo = [f for f in dir_list if "ckpt" in f][0]
+    checkpoint_file_stereo=os.path.join(experiment_path_stereo,modelfile)
     model_stereo = model_cls.load_from_checkpoint(checkpoint_file_stereo)
     model_stereo.model.roi_heads.score_thresh = 0.65
     experiment_path_charges = "./models/charges_model/"
-    dir_list = os.listdir(experiment_path_charges)
-    dir_list = [os.path.join(experiment_path_charges,f) for f in dir_list]
-    dir_list.sort(key=os.path.getctime, reverse=True)
-    checkpoint_file_charges = [f for f in dir_list if "ckpt" in f][0]
+    #dir_list = os.listdir(experiment_path_charges)
+    #dir_list = [os.path.join(experiment_path_charges,f) for f in dir_list]
+    #dir_list.sort(key=os.path.getctime, reverse=True)
+    #checkpoint_file_charges = [f for f in dir_list if "ckpt" in f][0]
+    checkpoint_file_charges=os.path.join(experiment_path_charges,modelfile)
     model_charge = model_cls.load_from_checkpoint(checkpoint_file_charges)
     model_charge.model.roi_heads.score_thresh = 0.65