search

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.
https://qiita.com/PINTO
MIT License
3.49k stars 566 forks source link

CreStereo onnx model conversion to tensorrt version #347

Closed waveajay closed 1 year ago

waveajay commented 1 year ago

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

PINTO0309 commented 1 year ago

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