Suppress duplicate YOLO tile detections

2026-07-09 10:55:45 +02:00
parent 638534f011
commit 3f5ca4c85e
24 changed files with 266 additions and 39 deletions
@@ -37,4 +37,5 @@ YOLO_DEVICE=cpu
 YOLO_IMAGE_SIZE=640
 YOLO_MAX_TILES=100
 YOLO_MAX_DETECTIONS=1000
+YOLO_DUPLICATE_IOU_THRESHOLD=0.5
 YOLO_BATCH_SIZE=1
@@ -31,6 +31,7 @@ YOLO_DEVICE="${YOLO_DEVICE:-cpu}"
 YOLO_IMAGE_SIZE="${YOLO_IMAGE_SIZE:-640}"
 YOLO_MAX_TILES="${YOLO_MAX_TILES:-100}"
 YOLO_MAX_DETECTIONS="${YOLO_MAX_DETECTIONS:-1000}"
+YOLO_DUPLICATE_IOU_THRESHOLD="${YOLO_DUPLICATE_IOU_THRESHOLD:-0.5}"
 YOLO_BATCH_SIZE="${YOLO_BATCH_SIZE:-1}"

 install_dockerman_metadata() {
@@ -92,6 +93,7 @@ docker run -d \
  -e YOLO_IMAGE_SIZE="$YOLO_IMAGE_SIZE" \
  -e YOLO_MAX_TILES="$YOLO_MAX_TILES" \
  -e YOLO_MAX_DETECTIONS="$YOLO_MAX_DETECTIONS" \
+  -e YOLO_DUPLICATE_IOU_THRESHOLD="$YOLO_DUPLICATE_IOU_THRESHOLD" \
  -e YOLO_BATCH_SIZE="$YOLO_BATCH_SIZE" \
  -v "${GEOINTEL_POSTGIS_DATA_PATH}:/var/lib/postgresql/data" \
  -v "${GEOINTEL_STORAGE_PATH}:/app/storage" \