![](https://img-blog.csdnimg.cn/20201014180756928.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
pytorch
日暮乡关何处是
这个作者很懒,什么都没留下…
展开
-
onnx.export报警告:WARNING: The shape inference of prim::Constant type is missing...解决方法
在将pytorch模型转换为onnx模型时报警:```WARNING: The shape inference of prim::Constant type is missing, so it may result in wrong shape inference for the exported graph. Please consider adding it in symbolic function.```经过模块尝试发现是`F.interpolate`函数(换成F.upsmple也报)原创 2022-01-14 10:26:59 · 12774 阅读 · 5 评论 -
安装torch2trt plugins报错解决方法:
错误描述:RuntimeError: Error compiling objects for extension解决方法:install with plugins · Issue #315 · NVIDIA-AI-IOT/torch2trt · GitHub为了防止有人没办法访问github,此处放出原文内容:```I solved it;you must modify thesetup.pyfile, let it known where is you tensorRT path of i.原创 2022-01-13 00:50:56 · 436 阅读 · 0 评论 -
python 常用的保存图像的函数输入图像的范围
如果图像输入范围错误,则会导致保存的图像显示错误,比如全黑。cv2.imwrite()输入图像范围在0~255之间。plt.imsave();输入图像范围在0~1之间。io.imsave();输入图像范围在0~255之间。原创 2020-12-23 16:18:49 · 922 阅读 · 0 评论 -
pytorch跑网络坑总结
学习神经网络有一年多了,终于终于开始跑网络了。此处记录自己因为不细心和对pytorch理解不够导致的一些错误,防止后面再犯类似错误。关于pytorch的transforms.RandomCrop和transforms.RandomHorizontalFlip由于做的课题是图像增强一类,因而数据和对应的l标签都是图像,在使用上述两个函数进行图像增强的过程中,发现经过transform处理后的图像和标签并不匹配,原因是因为上述函数有个随机因子。解决方法:直接取消图像增强自己写图像增强函数,然后原创 2020-12-19 17:34:40 · 440 阅读 · 1 评论 -
C++部署pytorch模型(下)
在之前的文章中分别介绍了利用pytorch实现一个简单的分类器和利用利用opencv加载训练好的模型和网络,今天开始介绍利用pytorch官方提供的LibTorch加载训练好的模型和网络。利用pytorch官方提供的LibTorch加载训练好的模型和网络参考链接:windows+VS2019+PyTorchLib配置使用攻略C++调用pytorch,LibTorch在win10下的vs配置和cmake的配置在C ++中加载TORCHSCRIPT模型官网链接此处首先说明一下将pytroch保存为T原创 2020-08-20 18:40:56 · 1786 阅读 · 0 评论 -
C++部署pytorch模型(中)
在C++部署pytorch模型(上)中我们已经将所有准备工作做好,今天开始利用opencv部署网络。利用opencv加载训练好的模型和网络参考链接:OpenCV4.0 运行快速风格迁移(Torch)opencv官方文档OpenCV加载Pytorch模型出现Unsupported Lua type 解决方法将模型从PYTORCH导出到ONNX并使用ONNX RUNTIME运行(官网链接)这个方法算是一个比较常用的方法,而且可以用到许多深度学习框架上面:根据opencv官方文档中的说明,可以支持原创 2020-08-20 17:54:21 · 1746 阅读 · 5 评论 -
C++部署pytorch模型(上)
文章目录C++部署pytorch模型(上)前言训练一个简单的pytorch网络C++部署pytorch模型(上)前言项目需要将pytorch训练好的网络用c++调用,在正式开始项目之前,在网上查了各种资料,共有三种实现方法:直接将网络从最基础的CNN模块用C++实现;将网咯模型和参数保存,然后使用opencv的DNN模块加载,这个方法tensorflow、torch等其他网络架构也能用,具体包含哪些下文会给出;使用pytorch官网提供的c++接口:LibTorch。其原理也是保存网络模型和参原创 2020-08-20 17:52:04 · 1678 阅读 · 0 评论 -
pytorch错误提示和应对方法总结(不定时更新)
本文为总结自己在调试pytorch程序中遇到的问题和解决方法Linear()层维度不匹配错误提示:size mismatch, m1: [20 x 12800], m2: [15488x 784] at…解决方法:nn.Linear(15488, 784)时报错,因为输入的矩阵维度为12800,因此将语句改为nn.Linear(12800, 784)即可。m1对应的输入x的维度,m2为权重w的维度。未完待续。。。ps:希望这篇文章越短越好。...原创 2020-07-19 09:34:31 · 493 阅读 · 0 评论