pytorch1.4 导出模型问题记录

在pytorch1.4,使用torch.jit.trace方法导出YOLO3模型时,执行:traced_script_module=torch.jit.trace(model,example) traced_script_module.save("yolov3.pt")
保存后,在C++环境下进行载入测试,载入的模型参数都正确,不知什么原因输出却不对。

最后将尝试将保存模型名称后缀改为.pth后traced_script_module.save("yolov3.pth"),输出正常。

2020.8.18 更新——————————————————————
发现网上现有的Yolov3 pytorch版在打包中多多少少都有一些问题,因此把我测试过的Pytorch 版Yolov3分享出来(不包含非极大值抑制,仅有前向传播,预处理也是直接resize,并非原码中的等比缩放贴在画布上,目的仅仅是做libtorch的测试,准备有时间了再写),已经上传到CSDN,c++参考的网上并做了一些修改,写的一般兄弟们凑活着看吧,权重文件可以在官网下载。我上传的里边对github下载的原始版本有一些修改(源网址上传文件readme里有),主要是将模型生成中使用numpy运算的几句全改成torch运算,以及测试时偷懒不想写c++的一些东西,无关大局。(https://download.csdn.net/download/weixin_45650404/12719737) 这尼玛分涨的太离谱了,需要的私聊留邮箱,我直接发。

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 18
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 18
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值