要在视觉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,自己添加了类别编号。
测试结果:
先加了概率,后面类别可以用coco.names文件读入,通过编号对应的类别输出即可,我的方法是按照将coco.names存在一个vector中,然后putext添加不同的vector元素。
ps:
1.cv::putText的api:
2.cv::rectange(主要想试试矩形填充)