perf(uav): exclude unused classes, lower threshold for vss to allow m…

…ore rare detections, and cluster everything
mbari-org · Oct 25, 2024 · c7d12b6 · c7d12b6
1 parent aefad28
commit c7d12b6
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 2 deletions.
diff --git a/aipipeline/projects/uav/config/config.yml b/aipipeline/projects/uav/config/config.yml
@@ -35,7 +35,7 @@ sdcat:
 
 vss:
   project: "901902-uavs"
-  threshold: 0.5
+  threshold: 0.1
   model: "google/vit-base-patch16-224"
 
 tator:

diff --git a/aipipeline/projects/uav/config/sdcat.ini b/aipipeline/projects/uav/config/sdcat.ini
@@ -3,7 +3,7 @@
 # Settings related to clustering
 ########################################################################
 # The minimum score of a detection to be considered for clustering; set to 0 to keep all detections
-min_score = 0.0
+min_score = -1
 # Location of the data to cluster - only used if day/night filtering is enabled
 # Monterey Bay
 latitude = 36.7253

diff --git a/aipipeline/projects/uav/load_sdcat_pipeline.py b/aipipeline/projects/uav/load_sdcat_pipeline.py
@@ -79,6 +79,10 @@ def process_mission(element) -> str:
         "--version",
         version,
         "--exclude",
+        "Poop",
+        "--exclude",
+        "Batray",
+        "--exclude",
         "Unknown",
     ]