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.53k stars 568 forks source link

Crestereo model conversion support #372

Open mantioh opened 11 months ago

mantioh commented 11 months ago

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