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 model conversion support

mantioh opened this issue · comments

Issue Type

Support, Feature Request

OS

Ubuntu

OS architecture

x86_64

Programming Language

Python

Framework

PyTorch, ONNX

Model name and Weights/Checkpoints URL

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

Description

Hai, I have converted CreStereo pytorch model to onnx using the below script

code

but im getting the different output resolution when compared to the models whatever uploaded in the repo

Model in the repo:
Original_image

Model converted using the script shown above:

Converted

Can you help resolving this, what might be the possible issue or is it possible to provide the code snippet you used for model conversion.

Relevant Log Output

No response

URL or source code for simple inference testing code

No response