YOLO-V3 实时目标检测

YOLO-V3 实时目标检测

官网：YOLO: Real-Time Object Detection

下载源码

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

其中 cfg/ 目录下有目前 yolov1，yolov2，yolov3 的配置。

下载模型

完整模型（ 237 MB ）：wget https://pjreddie.com/media/files/yolov3.weights

轻量模型（ 功能较差 ）：wget https://pjreddie.com/media/files/yolov3-tiny.weights

运行 DEMO

图片目标检测：

$ ./darknet detect cfg/yolov3.cfg yolov3.weights data/dog.jpg

layer     filters    size              input                output
    0 conv     32  3 x 3 / 1   608 x 608 x   3   ->   608 x 608 x  32  0.639 BFLOPs
    1 conv     64  3 x 3 / 2   608 x 608 x  32   ->   304 x 304 x  64  3.407 BFLOPs
    2 conv     32  1 x 1 / 1   304 x 304 x  64   ->   304 x 304 x  32  0.379 BFLOPs
    3 conv     64  3 x 3 / 1   304 x 304 x  32   ->   304 x 304 x  64  3.407 BFLOPs
    ...
  106 yolo
Loading weights from yolov3.weights...Done!
data/dog.jpg: Predicted in 31.758017 seconds.
dog: 100%
truck: 92%
bicycle: 99%

必须下载模型。

检测结果保存在本地 predictions.jpg 中。

测试图：