Jetson Nano使用yolov4-Tiny进行目标检测

1、安装darknet(AlexeyAB版本)

cd test/src/
git clone https://github.com/AlexeyAB/darknet.git

2、修改Makefile配置

使用CUDA和OPENCV进行编译
修改前:
在这里插入图片描述修改后:
在这里插入图片描述去掉52行的注释:
在这里插入图片描述

3、编译

cd darknet/
make

如果报错报错找不到nvcc,先查看nvcc文件是否存在:

cd /usr/local/cuda/bin
ls

在这里插入图片描述可知nvcc文件存在,报错是由于系统路径问题,vim .bashrc进入文件,添加对应路径,保存后再source ~/.bashrc

export PATH=$PATH:/usr/local/cuda-10.2/bin

再重新编译:

cd test/src/darknet
make

4、下载权重文件

去下方的网址下载已训练好的模型权重文件yolov4-tiny.weights,放入darknet文件夹中进行测试。

https://github.com/AlexeyAB/darknet/releases

5、使用yolov4-tiny.weights进行测试

5.1 图片测试

# detector test图片作为输入,使用经典的dog.jpg测试
./darknet detector test cfg/coco.data cfg/yolov4-tiny.cfg ./yolov4-tiny.weights data/dog.jpg

5.2 视频测试

./darknet detector demo cfg/coco.data cfg/yolov4-tiny.cfg ./yolov4-tiny.weights data/street.mp4

5.3 摄像头测试

# 检查是否有摄像头
ls /dev/ 
# 用摄像头进行测试
./darknet detector demo cfg/coco.data cfg/yolov4-tiny.cfg ./yolov4-tiny.weights /dev/video0

在这里插入图片描述

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值