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

Problem with StridedSlice #84

Closed beekeeper23 closed 2 years ago

beekeeper23 commented 2 years ago

Hello, I had some problems with the StridedSlice layer. I tried to slice the vector [16, 16, 6, 3] to [16, 16, 2, 3] by using as begin vector the [0, 0, 2, 0] , as end vector the [0, 0, 4, 0] and stride vector the [1, 1, 1, 1] but I received a vector with a size of [16, 2, 6, 3]. Can anyone help me?

stride

PINTO0309 commented 2 years ago

You can modify the attributes and weights as much as you want. https://github.com/PINTO0309/openvino2tensorflow#6-7-replace-weights-or-constant-values-in-const-op-and-add-transpose-or-reshape-or-cast-just-beforeafter-the-operation-specified-by-layer_id

beekeeper23 commented 2 years ago

Thanks for your comment. I tried this change but I think that the problem is on const layers.

PINTO0309 commented 2 years ago

Please read the README carefully. You can do almost anything you want.

PINTO0309 commented 2 years ago

Closed due to lack of progress.