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