Zoedepth ONNX conversion Script

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.43k stars 560 forks source link

Zoedepth ONNX conversion Script #384

Closed MesaMitch closed 6 months ago

MesaMitch commented 6 months ago

Issue Type

Support, Documentation Feature Request

OS

Ubuntu

OS architecture

x86_64

Programming Language

Python

Framework

PyTorch

Model name and Weights/Checkpoints URL

Zoedepth

Description

Conversion script for Zoedepth to ONNX

Relevant Log Output

No response

URL or source code for simple inference testing code

No response

PINTO0309 commented 6 months ago

ZoeDepth.zip

MesaMitch commented 6 months ago

Thank you!

MesaMitch commented 6 months ago

Please add the environment for onnx conversion it'd be really helpful, its throwing- AttributeError:'Tensor' object has no attribute 'astype'