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"