darknet安装与使用

darknet是一个开源的神经网络架构,使用C和CUDA编写,支持CPU和GPU计算.darknet实现RNN,ImageNet classification,yolo等网络机构,具体可以参考:https://pjreddie.com/darknet/.本文简单介绍darknet的安装和使用.

下载和编译darknet:

git clone https://github.com/pjreddie/darknet
cd darknet
make

下载已经训练好的yolo模型:

wget https://pjreddie.com/media/files/yolo.weights

测试:

./darknet detect cfg/yolo.cfg yolo.weights data/dog.jpg
或者./darknet detector test cfg/coco.data cfg/yolo.cfg yolo.weights data/dog.jpg

测试结构如下:

yolo默认设置只显示阈值大于0.25的检测,阈值可以通过thresh参数设定:

./darknet detect cfg/yolo.cfg yolo.weights data/dog.jpg -thresh 0