initial commit
This commit is contained in:
commit
0bafbee31f
54 files changed
+6317
No files matched your search
@@ -0,0 +1,34 @@
|
||||
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
|
||||
Reference in new issue
Block a user