ByteTrack/docker-run.sh at main · Salmon-Computer-Vision/ByteTrack · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
#!/usr/bin/env bash
name=$1
input=$2

# Variables
prefix=river
fps=10
raspi_ip=192.168.1.98
encoding=-5 # Set -5 if using h265
timezone="GMT+8"

# Constants
docker_image=masami.hopto.org:23411/bytetrack:dev
bytetrack_home=/home/salmonjetson/ByteTrack
workspace=/ByteTrack/YOLOX_outputs

mkdir -p "${bytetrack_home}"/YOLOX_outputs/track_outputs
sudo chattr +i "${bytetrack_home}"/YOLOX_outputs/track_outputs

# Run ByteTrack on input outputting to the YOLOX_outputs folder into the $prefix folder
sudo docker run -i --rm --runtime nvidia \
    -v ${bytetrack_home}/pretrained:/ByteTrack/pretrained \
    -v ${bytetrack_home}/datasets:/ByteTrack/datasets \
    -v ${bytetrack_home}/YOLOX_outputs:/ByteTrack/YOLOX_outputs \
    -v /tmp/.X11-unix/:/tmp/.X11-unix:rw \
    -v /home/salmonjetson/.ssh:/home/user/.ssh \
    -w "$workspace" \
    --device /dev/video0:/dev/video0:mwr \
    --net=host \
    --name "$name" \
    -e XDG_RUNTIME_DIR=$XDG_RUNTIME_DIR \
    -e DISPLAY=$DISPLAY \
    --privileged \
    $docker_image \
    ../tools/run_bytetrack.sh "${input}" "$prefix" $fps $raspi_ip $encoding $timezone