Cleaned up after reimage of metadata cpu

hdr-bgnn · Nov 14, 2022 · fa37ac2 · fa37ac2
1 parent e5ef5e4
commit fa37ac2
Show file tree

Hide file tree

Showing 7 changed files with 7 additions and 12 deletions.
diff --git a/check_labels.py b/check_labels.py
@@ -55,7 +55,7 @@ def init_model():
     # was 5 when I trained current model so has to stay 5 unless retrained
     cfg.MODEL.ROI_HEADS.NUM_CLASSES = 5
     cfg.MODEL.WEIGHTS = os.path.join(
-        cfg.OUTPUT_DIR, "model_final.pth.ocr")
+        cfg.OUTPUT_DIR, "model_final_ocr.pth")
     cfg.MODEL.ROI_HEADS.SCORE_THRESH_TEST = 0.3
     predictor = DefaultPredictor(cfg)
     return predictor

diff --git a/config/training_data.json.ocr b/config/training_data.json.ocr
diff --git a/config/training_data.json.orig b/config/training_data.json.orig
diff --git a/config/training_data.json.idigbio → config/training_data_idigbio.json b/config/training_data.json.idigbio → config/training_data_idigbio.json
diff --git a/config/training_data_ocr.json b/config/training_data_ocr.json
@@ -0,0 +1,3 @@
+{
+    "inhs_images_smaller/": ["1_labels.json"]
+}
diff --git a/config/training_data.json → config/training_data_orig.json b/config/training_data.json → config/training_data_orig.json
diff --git a/train_model_labels.py b/train_model_labels.py
@@ -30,10 +30,10 @@ def visualize_input(metadata, count):
 
 def main():
     prefix = open('config/overall_prefix.txt').readlines()[0].strip()
-    conf = json.load(open('config/training_data.json'))
+    conf = json.load(open('config/training_data_ocr.json'))
     metadata = None  # Need it in outer block for reuse
     train = []
-    test_images = f'{prefix}full_imgs/'
+    test_images = f'{prefix}inhs_images_smaller/'
 
     for img_dir in conf.keys():
         ims = f'{prefix}{img_dir}'
@@ -77,7 +77,7 @@ def main():
     trainer.resume_or_load(resume=True)
     trainer.train()
 
-    cfg.MODEL.WEIGHTS = os.path.join(cfg.OUTPUT_DIR, "enhance_model_final.pth")
+    cfg.MODEL.WEIGHTS = os.path.join(cfg.OUTPUT_DIR, "ocr_model_NEW.pth")
     # set the testing threshold for this model
     cfg.MODEL.ROI_HEADS.SCORE_THRESH_TEST = 0.1
     predictor = DefaultPredictor(cfg)