c++调用pytorch libtorch (YoloV3实战篇)
1.前言
此blog主要介绍本人在使用yolov3模型进行c++ libtorch 调试过程中遇到的一些问题。
Yolo-v3模型的介绍可参考深入拆解YOLO_V3,这里仍旧使用的是GitHub eriklindernoren/PyTorch-YOLOv3。
c++调用pytorch libtorch基础介绍可参考https://blog.csdn.net/WANGWUSHAN/article/details/118020188。该blog中包含了一个简单的demo,方便初学者快速入门。
2.直接使用仓库源码c++ libtorch输出结果与pytorch结果差别大
在custom的一个数据集上效果如下:
可见,直接使用源码通过torch.jit.trace生成的模型有点问题。可问题具体处在什么地方呢?