Jetson Nano是Nvidia在TX2和Xavier获得成功后推出的低配版GPU运算平台。测一下darknet版本yolov4-tiny在nano上的性能。
下载
git clone https://github.com/AlexeyAB/darknet.git
编译
- 修改Makefile
cd darknet
vim Makefile
令GPU=1, CUDNN=1, OPENCV=1
52行附近,取消注释
77行附近,修改NVCC的值
- 开始编译
make -j4
下载权重文件
wget https://github.com/AlexeyAB/darknet/releases/download/darknet_yolo_v4_pre/yolov4-tiny.weights
测试
./darknet detector test cfg/coco.data cfg/yolov4-tiny.cfg yolov4-tiny.weights data/horses.jpg
图片的大小为773*512,用时大概1000ms。