-
Notifications
You must be signed in to change notification settings - Fork 3
/
openpose.sh
executable file
·44 lines (40 loc) · 975 Bytes
/
openpose.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
if test "$#" -ne 2; then
echo "Expected two keywords: \$data_dir \$output_dir"
exit 1
fi
if [ ! -d "$1" ]; then
echo "Input directory '$1' does not exist! (exit)"
exit 1
fi
if [ ! -d "$2" ]; then
echo "Output directory '$2' does not exist!"
read -p "\tcreate it [y/n]? " yn
if [ "$yn" == "y" ]; then
mkdir "$2"
else
echo "(exit)"
exit 1
fi
fi
DOCKER_VERSION=$(docker version --format '{{.Server.Version}}')
echo "docker: $DOCKER_VERSION"
if [[ $DOCKER_VERSION == 19* ]]; then
docker run\
--gpus all\
--privileged\
--name='openpose_instance_generate'\
-v "$1":/home/user/data\
-v "$2":/home/user/output\
--rm\
-it jutanke/openpose\
/bin/bash exec_img.sh
else
nvidia-docker run\
--privileged\
--name='openpose_instance_generate'\
-v "$1":/home/user/data\
-v "$2":/home/user/output\
--rm\
-it jutanke/openpose\
/bin/bash exec_img.sh
fi