Open3

face-recognition-resnet100-arcface-onnx

git clone -b 2021.4.1 https://github.com/openvinotoolkit/open_model_zoo.git
cd open_model_zoo/tools/downloader

python3 downloader.py --name face-recognition-resnet100-arcface-onnx
python3 converter.py --name face-recognition-resnet100-arcface-onnx
cd public/face-recognition-resnet100-arcface-onnx
mv arcfaceresnet100-8.onnx arcfaceresnet100-8_org.onnx
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 \
ghcr.io/pinto0309/openvino2tensorflow:latest
H=112
W=112
MODEL=face-recognition-resnet100-arcface-onnx
openvino2tensorflow \
--model_path FP32/${MODEL}.xml \
--output_saved_model \
--output_pb \
--output_no_quant_float32_tflite \
--output_dynamic_range_quant_tflite \
--output_weight_quant_tflite \
--output_float16_quant_tflite \
--output_integer_quant_tflite \
--output_integer_quant_typ 'uint8' \
--string_formulas_for_normalization 'data / 255' \
--output_tfjs \
--output_coreml
openvino2tensorflow \
--model_path FP32/${MODEL}.xml \
--output_tftrt_float32
openvino2tensorflow \
--model_path FP32/${MODEL}.xml \
--output_tftrt_float16
mv saved_model ${MODEL}_${H}x${W}
rm -rf saved_model

openvino2tensorflow \
--model_path FP32/${MODEL}.xml \
--output_saved_model \
--output_pb \
--output_onnx \
--onnx_opset 11 \
--keep_input_tensor_in_nchw
onnx2trt ${MODEL}_${H}x${W}/model_float32.onnx \
-o ${MODEL}_${H}x${W}/${MODEL}_${H}x${W}_rtx3070.trt \
-b 1 \
-d 16 \
-v
ログインするとコメントできます