41 lines
1.1 KiB
Bash
Executable File
41 lines
1.1 KiB
Bash
Executable File
#!/usr/bin/env sh
|
|
set -eu
|
|
|
|
PROJECT_DIR="/opt/people-flow"
|
|
CONFIG_TEMPLATE="${PROJECT_DIR}/config/config.example.yaml"
|
|
CONFIG_PATH="${CONFIG_PATH:-${PROJECT_DIR}/config/local.yaml}"
|
|
OUTPUT_DIR="${OUTPUT_DIR:-${PROJECT_DIR}/outputs}"
|
|
RTSP_URL="${RTSP_URL:-}"
|
|
API_HOST="${API_HOST:-0.0.0.0}"
|
|
API_PORT="${API_PORT:-18082}"
|
|
|
|
mkdir -p "${OUTPUT_DIR}" "$(dirname "${CONFIG_PATH}")"
|
|
|
|
if [ ! -f "${CONFIG_PATH}" ]; then
|
|
cp "${CONFIG_TEMPLATE}" "${CONFIG_PATH}"
|
|
fi
|
|
|
|
python - "$CONFIG_PATH" "$RTSP_URL" "$OUTPUT_DIR" <<'PY'
|
|
from pathlib import Path
|
|
import sys
|
|
import yaml
|
|
|
|
config_path = Path(sys.argv[1])
|
|
rtsp_url = sys.argv[2]
|
|
output_dir = sys.argv[3]
|
|
|
|
raw = yaml.safe_load(config_path.read_text(encoding="utf-8")) or {}
|
|
runtime = raw.setdefault("runtime", {})
|
|
if rtsp_url:
|
|
runtime["rtsp_url"] = rtsp_url
|
|
runtime["output_dir"] = output_dir
|
|
yolo = raw.setdefault("yolo", {})
|
|
yolo.setdefault("model_path", "weights/yolo11n.pt")
|
|
config_path.write_text(
|
|
yaml.safe_dump(raw, allow_unicode=True, sort_keys=False),
|
|
encoding="utf-8",
|
|
)
|
|
PY
|
|
|
|
exec python main.py --config "${CONFIG_PATH}" manage-api --host "${API_HOST}" --port "${API_PORT}"
|