自定义博客皮肤VIP专享

*博客头图：

点击选择上传的图片

格式为PNG、JPG，宽度*高度大于1920*100像素，不超过2MB，主视觉建议放在右侧，请参照线上博客头图

请上传大于1920*100像素的图片！

博客底图：

点击选择上传的图片

图片格式为PNG、JPG，不超过1MB，可上下左右平铺至整个背景

栏目图：

点击选择上传的图片

图片格式为PNG、JPG，图片宽度*高度为300*38像素，不超过0.5MB

主标题颜色：

RGB颜色，例如：#AFAFAF

Hover：

RGB颜色，例如：#AFAFAF

副标题颜色：

RGB颜色，例如：#AFAFAF

预览取消提交

自定义博客皮肤

上一步保存

m0_46278347的博客

YuJL816 优快云认证博客专家优快云认证企业博客

码龄5年

暂无认证

1: 原创

141万+: 周排名

155万+: 总排名

3248: 访问

: 等级

51: 积分

2: 粉丝

1: 获赞

7: 评论

35: 收藏

私信

关注

分类专栏

pytorch 1篇

最新评论

在C++平台部署pytorch模型的两种方法
最佳运动员: [code=python] [code=python] import torch from models import Darknet model = Darknet('cfg/yolov3.cfg',608) model_weight_path = './weights/best.pt'#训练完模型位置 model.load_state_dict(torch.load(model_weight_path), False) model.cuda() model.eval() example = torch.rand(1, 3, 608, 608).cuda() traced_script_module = torch.jit.trace(model, example) traced_script_module.save("./weights/model.pt")#保存模型位置 [/code]
在C++平台部署pytorch模型的两种方法
最佳运动员: 请问这个怎么解决？ /home/xx405/anaconda3/envs/pytorch/bin/python /home/xx405/Downloads/YOLOv3_pytorch-master/zhuanhuan.py /home/xx405/Downloads/YOLOv3_pytorch-master/models.py:128: TracerWarning: Converting a tensor to a Python boolean might cause the trace to be incorrect. We can't record the data flow of Python values, so this value will be treated as a constant in the future. This means that the trace might not generalize to other inputs! if self.img_size != img_size: /home/xx405/Downloads/YOLOv3_pytorch-master/models.py:235: TracerWarning: Converting a tensor to a Python float might cause the trace to be incorrect. We can't record the data flow of Python values, so this value will be treated as a constant in the future. This means that the trace might not generalize to other inputs! self.nG = torch.FloatTensor([nG]).to(device) Traceback (most recent call last): File "/home/xx405/Downloads/YOLOv3_pytorch-master/zhuanhuan.py", line 9, in <module> traced_script_module = torch.jit.tra
在C++平台部署pytorch模型的两种方法
YuJL816 回复树脚丫: 我之前也碰到过这个问题，原因是在python中model.forward(imgs)返回有多个值，而再C++中torch::Tensor output = module.forward({ img_tensor }).toTensor();接收一个返回值，你把模型的forward改一下，让它return一个值，重新生成一个pt文件试试。
在C++平台部署pytorch模型的两种方法
YuJL816: 不需要
在C++平台部署pytorch模型的两种方法
zeta73: 这个python代码不需要用libtorch重写么

pytorch

关注

关注数：文章数：1 文章阅读量：3248 文章收藏量：35

作者: YuJL816

这个作者很懒，什么都没留下…

展开

在C++平台部署pytorch模型的两种方法

1、准备工作1、模型转换代码：import torchfrom model.FCN_Aux import FCN_AUXmodel = FCN_AUX(2)model_weight_path = './model/better.pth'#训练完模型位置model.load_state_dict(torch.load(model_weight_path))model.cuda()model.eval()example = torch.rand(1, 3, 224, 224).cuda()t

原创 2020-05-22 22:24:18 · 3248 阅读 · 7 评论

pytorch

作者: YuJL816

在C++平台部署pytorch模型的两种方法