From 073c90427188b8d8b437dbad05148cea3f7b1056 Mon Sep 17 00:00:00 2001 From: bw4sz Date: Wed, 27 Nov 2024 12:47:39 -0500 Subject: [PATCH] Add metadata --- conf/config.yaml | 9 +++++---- src/pipeline.py | 3 ++- 2 files changed, 7 insertions(+), 5 deletions(-) diff --git a/conf/config.yaml b/conf/config.yaml index 111e48a..2086828 100644 --- a/conf/config.yaml +++ b/conf/config.yaml @@ -62,11 +62,12 @@ pipeline_evaluation: image_dir: reporting: - image_dir: - report_dir: + image_dir: /blue/ewhite/b.weinstein/BOEM/reporting/samples + report_dir: /blue/ewhite/b.weinstein/BOEM/reporting/reports + metadata_dir: /blue/ewhite/b.weinstein/BOEM/reporting/metadata active_learning: - image_dir: /blue/ewhite/b.weinstein/BOEM/sample_flight/JPG_2024_Jan27/annotated + image_dir: /blue/ewhite/b.weinstein/BOEM/sample_flight/JPG_2024_Jan27 strategy: 'target-labels' n_images: 10 patch_size: 2000 @@ -82,7 +83,7 @@ active_learning: pool_limit: 500 active_testing: - image_dir: /blue/ewhite/b.weinstein/BOEM/sample_flight/JPG_2024_Jan27/annotated + image_dir: /blue/ewhite/b.weinstein/BOEM/sample_flight/JPG_2024_Jan27 strategy: 'random' n_images: 10 m: diff --git a/src/pipeline.py b/src/pipeline.py index 5e79693..71548bf 100644 --- a/src/pipeline.py +++ b/src/pipeline.py @@ -45,10 +45,11 @@ def run(self): annotated_images_dir=self.config.label_studio.annotated_images_dir, ) if new_annotations is None: - print("No new annotations, exiting") if self.config.force_upload: + print("No new annotations, but force_upload is set to True, continuing") self.skip_training = True else: + print("No new annotations, exiting") return None else: print(f"New annotations found: {len(new_annotations)}")