基于Jetson nano运行yolo3(Tiny版)

最新推荐文章于 2024-06-27 22:15:36 发布

wykup

最新推荐文章于 2024-06-27 22:15:36 发布

阅读量2.2k

点赞数 3

分类专栏： AI jetson nano linux

本文链接：https://blog.csdn.net/wyk71456/article/details/109169967

版权

AI 同时被 3 个专栏收录

12 篇文章 0 订阅

订阅专栏

linux

7 篇文章 0 订阅

订阅专栏

jetson nano

3 篇文章 0 订阅

订阅专栏

Jetson nano跑yolo3

yolo3是识别算法的明星，jetson nano是嵌入式平台里的小钢炮，不过它能能跑通yolo3吗？试过才知道！让GPU飞一会儿！

STEP1：下载Darknet 以及Yolov3的权重文件

git clone https://github.com/AlexeyAB/darknet

进入darknet文件夹，下载yolo3权重文件：

cd darknet
wget https://pjreddie.com/media/files/yolov3.weights
wget https://pjreddie.com/media/files/yolov3-tiny.weights

STEP2：配置GPU相关参数

sudo gedit Makefile

打开后配置参数(之前应该都是0):

GPU=1
CUDNN=1
OPENCV=1

STEP3：make一下

make

make前：
在这里插入图片描述
make后：多了obj文件夹和backup文件夹，以及可执行文件darknet

在这里插入图片描述

STEP5：最激动人心的一步，执行yolov3-tiny前向推理！

./darknet detector test cfg/coco.data cfg/yolov3-tiny.cfg yolov3-tiny.weights -ext_output dog.jpg

经典狗狗照：
在这里插入图片描述
马

估计是北京几环吧

STEP6：对视频文件检测

./darknet detector demo cfg/coco.data cfg/yolov3-tiny.cfg yolov3-tiny.weights -ext_output test.mp4

STEP7：摄像头检测

./darknet detector demo cfg/coco.data cfg/yolov3-tiny.cfg yolov3-tiny.weights "nvarguscamerasrc ! video/x-raw(memory:NVMM), width=1280, height=720, format=NV12, framerate=30/1 ! nvvidconv flip-method=2 ! video/x-raw, width=1280, height=720, format=BGRx ! videoconvert ! video/x-raw, format=BGR ! appsink"