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
334 stars 40 forks source link

yolov5s--- tpu compiler #53

Closed taloot closed 3 years ago

taloot commented 3 years ago

line 520, in _quantize return _mlir_quantize( File "/usr/local/lib/python3.8/dist-packages/tensorflow/lite/python/convert_phase.py", line 218, in wrapper raise error from None # Re-throws the exception. File "/usr/local/lib/python3.8/dist-packages/tensorflow/lite/python/convert_phase.py", line 208, in wrapper return func(*args, **kwargs) File "/usr/local/lib/python3.8/dist-packages/tensorflow/lite/python/convert.py", line 236, in mlir_quantize return wrap_toco.wrapped_experimental_mlir_quantize( File "/usr/local/lib/python3.8/dist-packages/tensorflow/lite/python/wrap_toco.py", line 47, in wrapped_experimental_mlir_quantize return _pywrap_toco_api.ExperimentalMlirQuantizeModel( RuntimeError: Failed to quantize:

i can't pypass the error becouse of i export the onnx with "--simplify" command.. if i didnt add simplify command it give layer error 5d at layer 407

PINTO0309 commented 3 years ago

There is too little information.