27 lines
547 B
Bash
27 lines
547 B
Bash
#!/usr/bin/env bash
|
|
set -euo pipefail
|
|
|
|
if [[ $# -lt 2 ]]; then
|
|
echo "Usage: $0 <rtsp_url> <host_output_dir>"
|
|
exit 1
|
|
fi
|
|
|
|
RTSP_URL="$1"
|
|
HOST_OUTPUT_DIR="$2"
|
|
|
|
mkdir -p "$HOST_OUTPUT_DIR"
|
|
|
|
docker run -d \
|
|
--name people-flow-rtsp \
|
|
--restart unless-stopped \
|
|
--network host \
|
|
--gpus all \
|
|
--shm-size 1g \
|
|
-v "$HOST_OUTPUT_DIR:/opt/people-flow/output" \
|
|
people-flow-rtsp:x86-cuda \
|
|
--config /opt/people-flow/configs/docker_x86_config.yaml \
|
|
--output-dir /opt/people-flow/output \
|
|
--device cuda:0 \
|
|
rtsp \
|
|
--input "$RTSP_URL"
|