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