35 lines
1.1 KiB
YAML
35 lines
1.1 KiB
YAML
services:
|
|
bird-cascade-batch:
|
|
build: .
|
|
image: bird-cascade-batch:latest
|
|
volumes:
|
|
- ./models:/app/models:ro
|
|
- ./input:/app/input:ro
|
|
- ./runs:/app/runs
|
|
environment:
|
|
ENABLE_DETECTION: "true"
|
|
ENABLE_CLASSIFICATION: "true"
|
|
ENABLE_DENSITY: "true"
|
|
DET_MODEL_PATH: /app/models/YNDC_20251010_yolov5_optimize.onnx
|
|
CLS_MODEL_PATH: /app/models/HHK_20260525_mmcls_optimize.onnx
|
|
DET_CLASSES_PATH: /app/models/YNDC_20251010_yolov5_optimize.txt
|
|
CLS_CLASSES_PATH: /app/models/HHK_20260525_mmcls_optimize.txt
|
|
DENSITY_MODEL_PATH: /app/models/bird_density.onnx
|
|
DET_IMG_SIZE: "640"
|
|
CLS_IMG_SIZE: "224"
|
|
DENSITY_IMG_SIZE: "0"
|
|
OUTPUT_DIR: /app/runs
|
|
TARGET_DET_CLASS_IDS: "0"
|
|
VIDEO_SAMPLE_INTERVAL: "2"
|
|
DEVICE: cpu
|
|
UNIFORM_SPECIES_COLOR: "false"
|
|
SMOOTH_VIDEO_BOXES: "true"
|
|
SMOOTH_HOLD_FRAMES: "3"
|
|
BOX_SMOOTH_ALPHA: "0.25"
|
|
FONT_PATH: /app/assets/fonts/NotoSansCJK-Regular.ttc
|
|
command:
|
|
- --input
|
|
- /app/input
|
|
- --output
|
|
- /app/runs
|