调试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,自己添加了类别编号。
测试结果:
在这里插入图片描述先加了概率,后面类别可以用coco.names文件读入,通过编号对应的类别输出即可,我的方法是按照将coco.names存在一个vector中,然后putext添加不同的vector元素。

ps:
1.cv::putText的api:

2.cv::rectange(主要想试试矩形填充)
在这里插入图片描述

  • 1
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值