-
Notifications
You must be signed in to change notification settings - Fork 0
/
test.sh
executable file
·68 lines (60 loc) · 1.76 KB
/
test.sh
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
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
#!/bin/bash
#####################################
##### User configured variables #####
ENABLE_REMOTE_GDB_DEBUGGER=0
TRACKER_ENABLED=0
GST_DEBUG_LEVEL=1
VIDEO="video.264"
TRACKER_TYPE="TLD"
INTERVAL=15
ENABLE_CUDA=1
CONFIDENCE=0.45
NMS=0.5
WIDTH=416
HEIGHT=416
CLASSES="$PWD/classes.txt"
CONFIG="$PWD/yolov4.cfg"
WEIGHTS="$PWD/yolov4.weights"
#####################################
#####################################
if [ -z "$1" ]
then
printf "\nNo video provided as an argument, trying \"%s\"\n\n" "$VIDEO"
sleep 1
else
VIDEO=$!
fi
ODO_GST_PATH="$PWD/release/gstreamer/"
LIB_PATH="$PWD/release/"
TRACKER=""
GDBSERVER=""
if [ $ENABLE_REMOTE_GDB_DEBUGGER -eq 1 ]; then
GDBSERVER="gdbserver 127.0.0.1:1234"
fi
if [ $TRACKER_ENABLED -eq 1 ]; then
TRACKER="odo_track type=$TRACKER_TYPE !"
else
INTERVAL=0
fi
if [ ! -f "$CLASSES" ]; then
printf "\nYolo classes file not found=%s" "$CLASSES"
printf "\nTry downloading them with \"get_yolo.sh\"\n\n"
exit 1
fi
if [ ! -f "$CONFIG" ]; then
printf "\nYolo config file not found=%s" "$CONFIG"
printf "\nTry downloading them with \"get_yolo.sh\"\n\n"
exit 1
fi
if [ ! -f "$WEIGHTS" ]; then
printf "\nYolo weights file not found=%s" "$WEIGHTS"
printf "\nDownload yolo files with \"get_yolo.sh\"\n\n"
exit 1
fi
export LD_LIBRARY_PATH=$LIB_PATH:$LD_LIBRARY_PATH
export GST_DEBUG=$GST_DEBUG_LEVEL
export GST_PLUGIN_PATH=$ODO_GST_PATH:$GST_PLUGIN_PATH
$GDBSERVER /usr/bin/gst-launch-1.0 filesrc location=$VIDEO \
! h264parse ! avdec_h264 ! videoconvert ! odo_detector interval=$INTERVAL cuda=$ENABLE_CUDA \
confidence=$CONFIDENCE nms=$NMS width=$WIDTH height=$HEIGHT lib_path=$LIB_PATH/libodo-lib-opencv-yolo.so \
net_classes=$CLASSES net_config=$CONFIG net_weights=$WEIGHTS ! $TRACKER odo_viz ! videoconvert ! autovideosink