NanoDet full updated

PINTO0309 · Aug 12, 2021 · c85498f · c85498f
1 parent 74f3ca6
commit c85498f
Show file tree

Hide file tree

Showing 9 changed files with 179 additions and 222 deletions.
diff --git a/072_NanoDet/04_integer_quantization.py b/072_NanoDet/04_integer_quantization.py
diff --git a/072_NanoDet/05_full_integer_quantization.py b/072_NanoDet/05_full_integer_quantization.py
diff --git a/072_NanoDet/06_edgetpu.txt b/072_NanoDet/06_edgetpu.txt
diff --git a/072_NanoDet/09_saved_model_to_coreml.py b/072_NanoDet/09_saved_model_to_coreml.py
diff --git a/072_NanoDet/10_saved_model_to_tfjs.txt b/072_NanoDet/10_saved_model_to_tfjs.txt
diff --git a/072_NanoDet/11_tensorrt_inf_test.py b/072_NanoDet/11_tensorrt_inf_test.py
diff --git a/072_NanoDet/convert_script.txt b/072_NanoDet/convert_script.txt
@@ -1,18 +1,116 @@
+xhost +local: && \
+docker run --gpus all -it --rm \
+-v `pwd`:/home/user/workdir \
+-v /tmp/.X11-unix/:/tmp/.X11-unix:rw \
+--device /dev/video0:/dev/video0:mwr \
+--net=host \
+-e XDG_RUNTIME_DIR=$XDG_RUNTIME_DIR \
+-e DISPLAY=$DISPLAY \
+--privileged \
+pinto0309/openvino2tensorflow:latest
+
+cd workdir
+
+
+MODEL=nanodet
+H=320
+W=320
+
+$INTEL_OPENVINO_DIR/deployment_tools/model_optimizer/mo.py \
+--input_model ${MODEL}_${H}x${W}.onnx \
+--data_type FP32 \
+--output_dir openvino/${H}x${W}/FP32
+$INTEL_OPENVINO_DIR/deployment_tools/model_optimizer/mo.py \
+--input_model ${MODEL}_${H}x${W}.onnx \
+--data_type FP16 \
+--output_dir openvino/${H}x${W}/FP16
+mkdir -p openvino/${H}x${W}/myriad
+${INTEL_OPENVINO_DIR}/deployment_tools/inference_engine/lib/intel64/myriad_compile \
+-m openvino/${H}x${W}/FP16/${MODEL}_${H}x${W}.xml \
+-ip U8 \
+-VPU_NUMBER_OF_SHAVES 4 \
+-VPU_NUMBER_OF_CMX_SLICES 4 \
+-o openvino/${H}x${W}/myriad/${MODEL}_${H}x${W}.blob
+
+
+MODEL=nanodet
+H=416
+W=416
+
+$INTEL_OPENVINO_DIR/deployment_tools/model_optimizer/mo.py \
+--input_model ${MODEL}_${H}x${W}.onnx \
+--data_type FP32 \
+--output_dir openvino/${H}x${W}/FP32
+$INTEL_OPENVINO_DIR/deployment_tools/model_optimizer/mo.py \
+--input_model ${MODEL}_${H}x${W}.onnx \
+--data_type FP16 \
+--output_dir openvino/${H}x${W}/FP16
+mkdir -p openvino/${H}x${W}/myriad
+${INTEL_OPENVINO_DIR}/deployment_tools/inference_engine/lib/intel64/myriad_compile \
+-m openvino/${H}x${W}/FP16/${MODEL}_${H}x${W}.xml \
+-ip U8 \
+-VPU_NUMBER_OF_SHAVES 4 \
+-VPU_NUMBER_OF_CMX_SLICES 4 \
+-o openvino/${H}x${W}/myriad/${MODEL}_${H}x${W}.blob
+
+
+
+
+
+MODEL=nanodet
+H=320
+W=320
 openvino2tensorflow \
-  --model_path onnx/openvino/nanodet_320x320/FP32/nanodet_320x320.xml \
-  --output_saved_model True \
-  --output_h5 True \
-  --output_pb True \
-  --output_no_quant_float32_tflite True \
-  --output_weight_quant_tflite True \
-  --output_float16_quant_tflite True
+--model_path openvino/${H}x${W}/FP32/${MODEL}_${H}x${W}.xml \
+--output_saved_model \
+--output_pb \
+--output_no_quant_float32_tflite \
+--output_weight_quant_tflite \
+--output_float16_quant_tflite \
+--output_integer_quant_tflite \
+--string_formulas_for_normalization 'data / 255' \
+--output_integer_quant_type 'uint8' \
+--output_tfjs \
+--output_tftrt \
+--output_coreml \
+--weight_replacement_config replace.json
+mv saved_model saved_model_${MODEL}_${H}x${W}
 
+openvino2tensorflow \
+--model_path openvino/${H}x${W}/FP32/${MODEL}_${H}x${W}.xml \
+--output_saved_model \
+--string_formulas_for_normalization 'data / 255' \
+--output_integer_quant_type 'uint8' \
+--output_edgetpu \
+--weight_replacement_config replace.json
+mv saved_model/model_full_integer_quant_edgetpu.tflite saved_model saved_model_${MODEL}_${H}x${W}
+rm -rf saved_model
+
+MODEL=nanodet
+H=416
+W=416
+openvino2tensorflow \
+--model_path openvino/${H}x${W}/FP32/${MODEL}_${H}x${W}.xml \
+--output_saved_model \
+--output_pb \
+--output_no_quant_float32_tflite \
+--output_weight_quant_tflite \
+--output_float16_quant_tflite \
+--output_integer_quant_tflite \
+--string_formulas_for_normalization 'data / 255' \
+--output_integer_quant_type 'uint8' \
+--output_tfjs \
+--output_tftrt \
+--output_coreml \
+--weight_replacement_config replace.json
+mv saved_model saved_model_${MODEL}_${H}x${W}
 
 openvino2tensorflow \
-  --model_path onnx/openvino/nanodet_416x416/FP32/nanodet_416x416.xml \
-  --output_saved_model True \
-  --output_h5 True \
-  --output_pb True \
-  --output_no_quant_float32_tflite True \
-  --output_weight_quant_tflite True \
-  --output_float16_quant_tflite True
+--model_path openvino/${H}x${W}/FP32/${MODEL}_${H}x${W}.xml \
+--output_saved_model \
+--string_formulas_for_normalization 'data / 255' \
+--output_integer_quant_type 'uint8' \
+--output_edgetpu \
+--weight_replacement_config replace.json
+mv saved_model/model_full_integer_quant_edgetpu.tflite saved_model saved_model_${MODEL}_${H}x${W}
+rm -rf saved_model
diff --git a/072_NanoDet/download.sh b/072_NanoDet/download.sh
@@ -1,8 +1,8 @@
 #!/bin/bash
 
-curl -sc /tmp/cookie "https://drive.google.com/uc?export=download&id=13u_o005EYKUcPhTMTijxrsgZHD3_ScZ_" > /dev/null
+curl -sc /tmp/cookie "https://drive.google.com/uc?export=download&id=1oER-UkW0qhzcFRUKCnsRttn7x_KFRn8T" > /dev/null
 CODE="$(awk '/_warning_/ {print $NF}' /tmp/cookie)"
-curl -Lb /tmp/cookie "https://drive.google.com/uc?export=download&confirm=${CODE}&id=13u_o005EYKUcPhTMTijxrsgZHD3_ScZ_" -o resources.tar.gz
+curl -Lb /tmp/cookie "https://drive.google.com/uc?export=download&confirm=${CODE}&id=1oER-UkW0qhzcFRUKCnsRttn7x_KFRn8T" -o resources.tar.gz
 tar -zxvf resources.tar.gz
 rm resources.tar.gz