Skip to content

Latest commit

 

History

History

zoe_depth

README.md

Zoe Depth

Input

(Image from NYU Depvh V2: https://drive.google.com/file/d/1AysroWpfISmm-yRFGBgFTrLy6FjQwvwP/view)

Output

  • ZoeD_M12_N

  • ZoeD_M12_NK

  • ZoeD_M12_K

Usage

To get the depth image from sample image, run the following comand

$ python3 zoe_depth.py

To specify ZoeDepth model, please use -a/--arch option. You can choose arch from ZoeD_M12_N, ZoeD_M12_NK and ZoeD_M12_K. The default arch is ZoeD_M12_K.

$ python3 zoe_depth.py -a <arch>

To use onnxruntime, set --onnx option

$ python3 zoe_depth.py --onnx

Reference

ZoeDepth

Framework

Pytorch

Model Format

ONNX opset = 17

Netron

ZoeD_M12_N.onnx.prototxt ZoeD_M12_NK.onnx.prototxt ZoeD_M12_K.onnx.prototxt