forked from zakaria/chicken-counting-sukawarna-det
proitlab
010f6e1493
- Add engine_utils for TensorRT compatibility verification, auto-recompilation from .pt models, and YAML auto-updates - Add mortality detection pipeline (mortality.py, test_run_mortality.sh, mortality_config.yaml) - Add multi-execution batch modes (parallel_processes, tensor_batching, hybrid) in batch_runner.py - Add daily test run automation scripts and video processing runners - Add dashboard REST API, live stream endpoints, and web UI templates - Clean up git tracking by ignoring __pycache__, .pyc, and build artifacts
50 lines
1.1 KiB
Bash
Executable File
50 lines
1.1 KiB
Bash
Executable File
#!/bin/bash
|
|
set -euo pipefail
|
|
|
|
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
|
|
export PYTHONPATH="$SCRIPT_DIR/src"
|
|
|
|
if [ -f "$SCRIPT_DIR/venv/bin/python" ]; then
|
|
PYTHON="$SCRIPT_DIR/venv/bin/python"
|
|
else
|
|
PYTHON="python3"
|
|
fi
|
|
|
|
CONFIG_FILE="${CONFIG_FILE:-configs/mortality_config.yaml}"
|
|
INPUT_PATH="${1:-}"
|
|
OUTPUT_DIR="${2:-}"
|
|
|
|
echo "=== Starting Whole-Image Mortality Chicken Detection ==="
|
|
echo "Config File: $CONFIG_FILE"
|
|
if [ -n "$INPUT_PATH" ]; then
|
|
echo "Input Path Override: $INPUT_PATH"
|
|
fi
|
|
if [ -n "$OUTPUT_DIR" ]; then
|
|
echo "Output Dir Override: $OUTPUT_DIR"
|
|
fi
|
|
|
|
CMD=($PYTHON -m chicken_counter.cli mortality --config "$CONFIG_FILE")
|
|
|
|
if [ -n "$INPUT_PATH" ]; then
|
|
CMD+=(--input "$INPUT_PATH")
|
|
fi
|
|
if [ -n "$OUTPUT_DIR" ]; then
|
|
CMD+=(--output-dir "$OUTPUT_DIR")
|
|
fi
|
|
if [ -n "${CONF:-}" ]; then
|
|
CMD+=(--conf "$CONF")
|
|
fi
|
|
if [ -n "${IOU:-}" ]; then
|
|
CMD+=(--iou "$IOU")
|
|
fi
|
|
if [ -n "${MIN_AREA:-}" ]; then
|
|
CMD+=(--min-area "$MIN_AREA")
|
|
fi
|
|
if [ -n "${DEDUPE_RADIUS:-}" ]; then
|
|
CMD+=(--dedupe-radius "$DEDUPE_RADIUS")
|
|
fi
|
|
|
|
"${CMD[@]}"
|
|
|
|
echo "=== Mortality Detection Completed ==="
|