repair_cls

app.py

@@ -108,7 +108,8 @@ CLS_MODEL_PATHS = OrderedDict({
     #"15shot_aug":   f"{MODELS_ROOT_DIR}/Classification/resnet50_hela_15_shot_aug_re",
     #"20shot_aug":   f"{MODELS_ROOT_DIR}/Classification/resnet50_hela_20_shot_aug_re",
 })
-CLS_CLASS_NAMES = ['dap', 'erdak', 'giant', 'gpp130', 'h4b4', 'mc151', 'nucle', 'phal', 'tfr', 'tubul']
+#CLS_CLASS_NAMES = ['dap', 'erdak', 'giant', 'gpp130', 'h4b4', 'mc151', 'nucle', 'phal', 'tfr', 'tubul']
+CLS_CLASS_NAMES = ['Nucleus', 'Endoplasmic Reticulum', 'Giantin', 'GPP130', 'Lysosomes', 'Mitochondria', 'Nucleolus', 'Actin', 'Endosomes', 'Microtubules']
 CLS_EXAMPLE_IMG_DIR = "example_images_cls"
 
 

prompts/basic_prompts.json

@@ -1,24 +1,24 @@
 [
+  "CCPs of COS-7",
+  "ER of COS-7",
+  "Microtubules of COS-7",
+  "F-actin of COS-7",
   "21MT1",
   "AGP of U2OS",
   "AU565",
   "Actin of HUVEC",
   "Actin of MCF-7",
-  "CCPs of COS-7",
   "DNA of U2OS",
   "E.coli",
-  "ER of COS-7",
   "ER of U2OS",
   "F-actin and plasma membrane of U2OS",
   "F-actin of BPAE",
-  "F-actin of COS-7",
   "GFP of Y6545",
   "GFP-tubulin or E-cadherin of caco2",
   "GOWT1 of mouse stem cells",
   "Golgi and Actin of U2OS",
   "H-NS-mScarlet-I of E.coli",
   "H2B-mCherry of Y6545",
-  "Microtubules of COS-7",
   "Microtubules of HUVEC",
   "Mitochondria of U2OS",
   "MreB of E.coli",