88 lines
2.6 KiB
Bash
Executable File
88 lines
2.6 KiB
Bash
Executable File
#!/bin/bash
|
|
# ============================================================================
|
|
# Saqr PPE Detection - Run on Local Laptop
|
|
# ============================================================================
|
|
#
|
|
# Usage:
|
|
# scripts/run_local.sh # webcam 0
|
|
# scripts/run_local.sh --source 1 # webcam 1
|
|
# scripts/run_local.sh --source video.mp4 # video file
|
|
# scripts/run_local.sh --gui # PySide6 GUI
|
|
# scripts/run_local.sh --detect # simple detection (no tracking)
|
|
#
|
|
# ============================================================================
|
|
|
|
set -e
|
|
|
|
HERE="$(cd "$(dirname "$0")" && pwd)"
|
|
PROJECT_ROOT="$(cd "$HERE/.." && pwd)"
|
|
cd "$PROJECT_ROOT"
|
|
|
|
SOURCE="0"
|
|
MODEL="saqr_best.pt"
|
|
CONF="0.35"
|
|
MODE="saqr" # saqr | gui | detect
|
|
HEADLESS=false
|
|
MAX_MISSING=90
|
|
MATCH_DIST=250
|
|
CONFIRM=5
|
|
|
|
while [[ $# -gt 0 ]]; do
|
|
case $1 in
|
|
--source) SOURCE="$2"; shift 2 ;;
|
|
--model) MODEL="$2"; shift 2 ;;
|
|
--conf) CONF="$2"; shift 2 ;;
|
|
--gui) MODE="gui"; shift ;;
|
|
--detect) MODE="detect"; shift ;;
|
|
--headless) HEADLESS=true; shift ;;
|
|
--max-missing) MAX_MISSING="$2"; shift 2 ;;
|
|
--match-distance) MATCH_DIST="$2"; shift 2 ;;
|
|
--confirm) CONFIRM="$2"; shift 2 ;;
|
|
*) echo "Unknown arg: $1"; exit 1 ;;
|
|
esac
|
|
done
|
|
|
|
if command -v conda &>/dev/null; then
|
|
source "$(conda info --base)/etc/profile.d/conda.sh" 2>/dev/null || true
|
|
conda activate AI_MSI_yolo 2>/dev/null || true
|
|
fi
|
|
|
|
echo "============================================"
|
|
echo " Saqr PPE Detection - Local Laptop"
|
|
echo "============================================"
|
|
echo " Mode : $MODE"
|
|
echo " Source : $SOURCE"
|
|
echo " Model : $MODEL"
|
|
echo " Conf : $CONF"
|
|
echo "============================================"
|
|
echo ""
|
|
|
|
HEADLESS_FLAG=""
|
|
if [ "$HEADLESS" = true ]; then
|
|
HEADLESS_FLAG="--headless"
|
|
fi
|
|
|
|
case $MODE in
|
|
saqr)
|
|
python -m saqr.apps.saqr_cli \
|
|
--source "$SOURCE" \
|
|
--model "$MODEL" \
|
|
--conf "$CONF" \
|
|
--max-missing "$MAX_MISSING" \
|
|
--match-distance "$MATCH_DIST" \
|
|
--status-confirm-frames "$CONFIRM" \
|
|
$HEADLESS_FLAG
|
|
;;
|
|
gui)
|
|
python -m saqr.gui.app \
|
|
--source "$SOURCE" \
|
|
--model "$MODEL"
|
|
;;
|
|
detect)
|
|
python -m saqr.apps.detect_cli \
|
|
--source "$SOURCE" \
|
|
--model "$MODEL" \
|
|
--conf "$CONF"
|
|
;;
|
|
esac
|