Commit 064e885f authored by Ukhu's avatar Ukhu
Browse files

Update benchmark download and convert

parent b7966865
FROM local/dockervino:phase1 as build
ARG downloader=/opt/intel/openvino/deployment_tools/tools/model_downloader/downloader.py
ARG optimizer=/opt/intel/openvino/deployment_tools/model_optimizer/mo.py
ENV downloader=${INTEL_OPENVINO_DIR}/deployment_tools/tools/model_downloader/downloader.py
ENV converter=${INTEL_OPENVINO_DIR}/deployment_tools/tools/model_downloader/converter.py
ENV optimizer=${INTEL_OPENVINO_DIR}/deployment_tools/model_optimizer/mo.py
ARG models=/root/openvino_models
WORKDIR $models
RUN python3 $downloader --name googlenet-v3 --output_dir $models
RUN python3 $optimizer --input_model ./public/googlenet-v3/inception_v3_2016_08_28_frozen.pb --output_dir ./ir/googlenet/v3/FP32 --data_type FP32
RUN python3 $optimizer --input_model ./public/googlenet-v3/inception_v3_2016_08_28_frozen.pb --output_dir ./ir/googlenet/v3/FP16 --data_type FP16
RUN python3 $converter --name googlenet-v3 --output_dir $models --precisions FP32,FP16
RUN python3 $downloader --name googlenet-v4-tf --output_dir $models
RUN python3 $optimizer --input_model ./public/googlenet-v4-tf/inception_v4.frozen.pb --output_dir ./ir/googlenet/v4/FP32 --data_type FP32
RUN python3 $optimizer --input_model ./public/googlenet-v4-tf/inception_v4.frozen.pb --output_dir ./ir/googlenet/v4/FP16 --data_type FP16
RUN python3 $converter --name googlenet-v4-tf --output_dir $models --precisions FP32,FP16
RUN python3 $downloader --name vgg16 --output_dir $models
RUN python3 $optimizer --input_model ./public/vgg16/vgg16.caffemodel --output_dir ./ir/vgg/16/FP32 --data_type FP32
RUN python3 $optimizer --input_model ./public/vgg16/vgg16.caffemodel --output_dir ./ir/vgg/16/FP16 --data_type FP16
RUN python3 $converter --name vgg16 --output_dir $models --precisions FP32,FP16
RUN python3 $downloader --name vgg19 --output_dir $models
RUN python3 $optimizer --input_model ./public/vgg19/vgg19.caffemodel --output_dir ./ir/vgg/19/FP32 --data_type FP32
RUN python3 $optimizer --input_model ./public/vgg19/vgg19.caffemodel --output_dir ./ir/vgg/19/FP16 --data_type FP16
#RUN python3 $downloader --name resnet-50 --output_dir $models
#RUN python3 $optimizer --input_model ./public/resnet-50/resnet-50.caffemodel --output_dir ./ir/resnet/v1/50/FP32 --data_type FP32
#RUN python3 $optimizer --input_model ./public/resnet-50/resnet-50.caffemodel --output_dir ./ir/resnet/v1/50/FP16 --data_type FP16
#RUN python3 $downloader --name resnet-101 --output_dir $models
#RUN python3 $optimizer --input_model ./public/resnet-101/resnet-101.caffemodel --output_dir ./ir/resnet/v1/101/FP32 --data_type FP32
#RUN python3 $optimizer --input_model ./public/resnet-101/resnet-101.caffemodel --output_dir ./ir/resnet/v1/101/FP16 --data_type FP16
RUN python3 $converter --name vgg19 --output_dir $models --precisions FP32,FP16
#RUN python3 $downloader --name resnet-152 --output_dir $models
#RUN python3 $optimizer --input_model ./public/resnet-152/resnet-152.caffemodel --output_dir ./ir/resnet/v1/152/FP32 --data_type FP32
#RUN python3 $optimizer --input_model ./public/resnet-152/resnet-152.caffemodel --output_dir ./ir/resnet/v1/152/FP16 --data_type FP16
RUN python3 $downloader --name resnet-50-tf --output_dir $models
RUN python3 $converter --name resnet-50-tf --output_dir $models --precisions FP32,FP16
COPY ./entrypoint.sh /
CMD ["/bin/bash"]
......@@ -12,49 +12,38 @@ lscpu
if [ "$device" == "CPU" ]; then
echo $stars
echo "Inception V3"
$benchmark -d $device -i $input -api $api -m $models/ir/googlenet/v3/FP32/inception_v3_2016_08_28_frozen.xml
$benchmark -d $device -i $input -api $api -m $models/public/googlenet-v3/FP32/googlenet-v3.xml
echo $stars
echo "Inception V4"
$benchmark -d $device -i $input -api $api -m $models/ir/googlenet/v4/FP32/inception_v4.frozen.xml
$benchmark -d $device -i $input -api $api -m $models/public/googlenet-v4-tf/FP32/googlenet-v4-tf.xml
echo $stars
echo "VGG16"
$benchmark -d $device -i $input -api $api -m $models/ir/vgg/16/FP32/vgg16.xml
$benchmark -d $device -i $input -api $api -m $models/public/vgg16/FP32/vgg16.xml
echo $stars
echo "VGG19"
$benchmark -d $device -i $input -api $api -m $models/ir/vgg/19/FP32/vgg19.xml
$benchmark -d $device -i $input -api $api -m $models/public/vgg19/FP32/vgg19.xml
echo $stars
#echo "Resnet 50"
#$benchmark -d $device -i $input -api $api -m $models/ir/resnet/v1/50/FP32/resnet-50.xml
#echo $stars
#echo "Resnet 101"
#$benchmark -d $device -i $input -api $api -m $models/ir/resnet/v1/101/FP32/resnet-101.xml
#echo $stars
#echo "Resnet 152"
#$benchmark -d $device -i $input -api $api -m $models/ir/resnet/v1/152/FP32/resnet-152.xml
#echo $stars
echo "Resnet 50"
$benchmark -d $device -i $input -api $api -m $models/public/resnet-50-tf/FP32/resnet-50-tf.xml
else
echo $stars
echo "Inception V3"
$benchmark -d $device -i $input -api $api -m $models/ir/googlenet/v3/FP16/inception_v3_2016_08_28_frozen.xml
$benchmark -d $device -i $input -api $api -m $models/public/googlenet-v3/FP16/googlenet
-v3.xml
echo $stars
echo "Inception V4"
$benchmark -d $device -i $input -api $api -m $models/ir/googlenet/v4/FP16/inception_v4.frozen.xml
echo $stars
$benchmark -d $device -i $input -api $api -m $models/public/googlenet-v4-tf/FP16/google
net-v4-tf.xml
echo $stars
echo "VGG16"
$benchmark -d $device -i $input -api $api -m $models/ir/vgg/16/FP16/vgg16.xml
$benchmark -d $device -i $input -api $api -m $models/public/vgg16/FP16/vgg16.xml
echo $stars
echo "VGG19"
$benchmark -d $device -i $input -api $api -m $models/ir/vgg/19/FP16/vgg19.xml
$benchmark -d $device -i $input -api $api -m $models/public/vgg19/FP16/vgg19.xml
echo $stars
#echo "Resnet 50"
#$benchmark -d $device -i $input -api $api -m $models/ir/resnet/v1/50/FP16/resnet-50.xml
#echo $stars
#echo "Resnet 101"
#$benchmark -d $device -i $input -api $api -m $models/ir/resnet/v1/101/FP16/resnet-101.xml
#echo $stars
#echo "Resnet 152"
#$benchmark -d $device -i $input -api $api -m $models/ir/resnet/v1/152/FP16/resnet-152.xml
#echo $stars
echo "Resnet 50"
$benchmark -d $device -i $input -api $api -m $models/public/resnet-50-tf/FP16/resnet-50
-tf.xml
fi
echo "Benchmarking done!"
......
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