pytorch生产部署

本文探讨了PyTorch模型的两种常见部署方式:利用libtorch直接部署和转换为ONNX模型后通过OpenCV调用。提到了在C++中部署PyTorch模型时遇到的问题,如cv::exception,并分享了多个关于PyTorch到ONNX转换的资源。还讨论了C#调用ONNX模型的可能性,引用了官方ONNXRuntime的C#示例和相关博客文章,以供参考。

目前看常用的部署方式有:

1.利用libtorch部署;2.转为onnx模型,然后再用opencv进行调用。

例如 C++部署pytorch模型(中) https://blog.csdn.net/cai1493105270/article/details/108129126

Pytorch训练的分割网络模型在OpenCV4.0/C++上部署 https://blog.csdn.net/ppCuda/article/details/103393679

但是在实际调用过程中发现dnn::readNetFromONNX会报错cv::exception

pytorch 转onnx文章(未实际运行测试):

https://github.com/bleakie/CenterMulti/blob/master/TensorRT/convert2onnx.py

https://github.com/onnx/models

https://github.com/onnx/models/blob/c977da62627932baa51558f3a5ccfb291e222b4c/vision

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值