导出zoedepth的onnx模型并基于gradio实现在线部署

发布时间:2024年01月17日

运行环境说明,其依赖: torch + python-opencv + onnxruntime + gradio + numpy + einops
导出onnx后运行环境依赖: python-opencv + onnxruntime + gradio + numpy + einops

1、 前置修改

1.1 ZoeDepth运行前准备

参考使用ZoeDepth生成深度估计图 ,下载项目:https://github.com/isl-org/ZoeDepth

同时参考该章节中的2.3修改代码,最终实现能在pytorch环境下进行推理

1.2 nn.functional.interpolate报错修改

报错信息如下所示:

File “C:\Users\Administrator\miniconda3\lib\site-packages\torch\nn\functional.py”, line 3924, in interpolate
raise TypeError(
TypeError: expected size to be one of int or Tuple[int] or Tuple[int, int] or Tuple[int, int, int], but got size with types [<class ‘numpy.int32’>, <class ‘numpy.int32’>]

将 C:\Users\Administrator\miniconda

文章来源:https://blog.csdn.net/a486259/article/details/135635237
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。