search

PINTO0309 / openvino2tensorflow

This script converts the ONNX/OpenVINO IR model to Tensorflow's saved_model, tflite, h5, tfjs, tftrt(TensorRT), CoreML, EdgeTPU, ONNX and pb. PyTorch (NCHW) -> ONNX (NCHW) -> OpenVINO (NCHW) -> openvino2tensorflow -> Tensorflow/Keras (NHWC/NCHW) -> TFLite (NHWC/NCHW). And the conversion from .pb to saved_model and from saved_model to .pb and from .pb to .tflite and saved_model to .tflite and saved_model to onnx. Support for building environments with Docker. It is possible to directly access the host PC GUI and the camera to verify the operation. NVIDIA GPU (dGPU) support. Intel iHD GPU (iGPU) support.
MIT License
338 stars 39 forks source link

[tflite] Yolact's output "prior" is missing. #37

Closed taisuke-tomida closed 3 years ago

taisuke-tomida commented 3 years ago

こんにちは　@PINTO0309 さん。

openvino2tensorflowを使わせていただいて、yolactをtflite化しました。
onnxでの推論はできました。
onnxとtfliteのモデルをNETRONで確認したところ、出力の[19248, 4]の部分がないことに気づきました。		 TFLite ONNX
image image

https://github.com/Ma-Dan/yolact/tree/onnx　 こちらのコードでtfliteの推論を試そうとしています。この出力がなくても推論はできるんでしょうか。
よろしくお願いします。