调试github上的c++版libtorch-yolov3

要在视觉SLAM工程中加入YOLOv3的目标检测,所以找到了这个代码。
主要借助与libtorch这个c++版的torch库api

环境:ubuntu18.04+CUDA10.1+cudnn7.5+libtorch1.5

1.CUDA10.1+cudnn7.5+libtorch1.5的安装就按照教程走就好了;
推荐的链接安装CUDA和cudnn
2.libtorch-yolov3的编译
不同的libtorh版本对应不同的api,所以在编译过程中按照issue的指示一个个修改即可;
推荐修改issue
3.这个编译出来只有box,自己添加了类别编号。
测试结果:
调试github上的c++版libtorch-yolov3_第1张图片先加了概率,后面类别可以用coco.names文件读入,通过编号对应的类别输出即可,我的方法是按照将coco.names存在一个vector中,然后putext添加不同的vector元素。

ps:
1.cv::putText的api:
调试github上的c++版libtorch-yolov3_第2张图片
2.cv::rectange(主要想试试矩形填充)
调试github上的c++版libtorch-yolov3_第3张图片

你可能感兴趣的:(调试github上的c++版libtorch-yolov3)