search

lpiccinelli-eth / UniDepth

Universal Monocular Metric Depth Estimation
Other
588 stars 47 forks source link

export to onnx #34

Closed miiddb closed 5 months ago

miiddb commented 5 months ago

Hello, Would it be possible to export the model into the .onnx format ?

lpiccinelli-eth commented 5 months ago

It is in our roadmap, we will release version2 an then the onnx exporter for it.

lpiccinelli-eth commented 5 months ago

It has been included in PR #38 which will release UniDepthV2 and .onnx exporting for V2.

You can check on the version2 branch the unidepth/models/unidepthv2/export.py file and documentation under assets/docs/V2_README.md to export our model to .onnx format. You can give it a try if you want and give us any feedback, thanks!

miiddb commented 5 months ago

thanks, it seems to work! Now i'm waiting for the smaller versions!

charleshuang880730 commented 3 months ago

thanks, it seems to work! Now i'm waiting for the smaller versions!

@miiddb Hi, I would like to ask if you could provide the converted ONNX model. I encountered errors during the conversion process and urgently need the ONNX model for testing. I would greatly appreciate your help. Thank you.