Giters

PINTO0309 / PINTO_model_zoo

A repository for storing models that have been inter-converted between various frameworks. Supported frameworks are TensorFlow, PyTorch, ONNX, OpenVINO, TFJS, TFTRT, TensorFlowLite (Float32/16/INT8), EdgeTPU, CoreML.

Home Page:https://qiita.com/PINTO

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

CreStereo onnx model conversion to tensorrt version

waveajay opened this issue · comments

commented

Issue Type

Support, Documentation Feature Request

OS

Windows, Ubuntu

OS architecture

x86_64

Programming Language

C++, Python

Framework

ONNX, TensorRT

Model name and Weights/Checkpoints URL

https://github.com/PINTO0309/PINTO_model_zoo/tree/main/284_CREStereo

Description

In the repository i could see two version of onnx i,e normal onnx and tensorrt version that is also onnx.
Can you provide the script/proceudre for converting normal onnx to tenssort.onnx

Relevant Log Output

No response

URL or source code for simple inference testing code

No response

commented

TensorRT does not accept 3 inputs for Einsum, so add another Einsum before Einsum with 3 inputs. Unfortunately, I have not kept any past resources on hand for more than a year. Good luck.

https://github.com/ibaiGorordo/CREStereo-Pytorch

image