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) 这尼玛分涨的太离谱了,需要的私聊留邮箱,我直接发。

你可能感兴趣的:(笔记)