Commit 8ab0f6fd authored by Juan Diego's avatar Juan Diego
Browse files

Instance segmentation

parent dd43f049
......@@ -23,6 +23,7 @@ rm -f ./xauthority ; cp $XAUTHORITY ./xauthority ;chmod 666 ./xauthority #else r
setupvars=/opt/intel/openvino/bin/setupvars.sh
examples_dir=/root/omz_demos_build/intel64/Release
models_dir=/root/openvino_models/ir
labels_file=/opt/intel/openvino/deployment_tools/open_model_zoo/data/dataset_classes/coco_80cl.txt
build_cmd="/opt/intel/openvino/deployment_tools/open_model_zoo/demos/build_demos.sh -DENABLE_PYTHON=on"
# Input: IPCam or USBCam
input=0
......@@ -32,8 +33,8 @@ input=0
device=CPU
######## CPU 3D Pose Estimation ###
if [ "$device" == CPU ]; then
app=/opt/intel/openvino/deployment_tools/open_model_zoo/demos/human_pose_estimation_3d_demo/python/human_pose_estimation_3d_demo.py
model=${models_dir}/public/human-pose-estimation-3d-0001/FP32/human-pose-estimation-3d-0001.xml
app=/opt/intel/openvino/deployment_tools/open_model_zoo/demos/instance_segmentation_demo/python/instance_segmentation_demo.py
model=${models_dir}/intel/instance-segmentation-security-0228/FP32/instance-segmentation-security-0228.xml
fi
###### NCS2 3D Pose Estimation #####
if [ "$device" == MYRIAD ]; then
......@@ -59,12 +60,12 @@ docker run $opts \
-ti \
--mount type=bind,source="$(pwd)"/../../open_model_zoo,target=/opt/intel/openvino/deployment_tools/open_model_zoo \
--mount type=bind,source="$(pwd)"/../demo/videos,target=/videos \
--privileged \
--net=host \
--env="setupvars=$setupvars" \
--env="examples_dir=$examples_dir" \
--env="models_dir=$models_dir" \
--env="build_cmd=$build_cmd" \
--env="labels_file=$labels_file" \
--env="DISPLAY=$display" \
--env="input=$input" \
--env="device=$device" \
......@@ -76,8 +77,7 @@ docker run $opts \
--env="agM=$agM" \
--env="emM=$emM" \
--env="hpM=$hpM" \
--device="/dev/video0:/dev/video0" \
--volume="/dev:/dev" \
--device="/dev/dri:/dev/dri" \
--volume="/tmp/.X11-unix:/tmp/.X11-unix" \
--env="PS1=$myname> "\
--publish-all=true \
......
......@@ -91,5 +91,7 @@ RUN python3 $optimizer \
--model_name mask_rcnn_inception_v2 \
--data_type FP16 --reverse_input_channels
# Download Instance Segmentation
RUN python3 $downloader --name instance-segmentation-security-0228 --output_dir $models/ir
CMD ["/bin/bash"]
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment