darknet优化经验-AlexeyAB大神经验

目录

• darknet优化经验
  • 1. AlexeyAB改进项
  • 2. Linux下编译选项
  • 3. 训练经验
  • 4. 提高检测效果
  • 5. 总结
  • 6. AlexeyAB大神改进

darknet优化经验

主要来自于：AlexeyAB 版本darknet

1. AlexeyAB改进项

• 提供window支持

• 相较于原版pjreddie版本darknet提高了训练速度

• 添加了二值化网络，XNOR(bit) ，速度快，准确率稍低https://github.com/AlexeyAB/darknet/blob/master/cfg/yolov3-tiny_xnor.cfg

• 提高7%经过将卷积层和BN层合并为一个（*_*）不太懂。

• 多GPU训练提高

• 修补了[reorg]层

• 添加了mAP, IOU，Precision-Recall计算

  darknet detector map...

• 能够在训练过程当中画loss图像

• 添加了根据本身数据集的anchor生成

• 提高视频检测，网络摄像头，opencv相关问题

• 提出了一个INT8的网络，提高了检测速度，可是准确率稍有降低

  https://github.com/AlexeyAB/yolo2_light

2. Linux下编译选项

• GPU=1 to build with CUDA to accelerate by using GPU (CUDA should be in /usr/local/cuda)
• CUDNN=1 to build with cuDNN v5-v7 to accelerate training by using GPU (cuDNN should be in /usr/local/cudnn)
• CUDNN_HALF=1 to build for Tensor Cores (on Titan V / Tesla V100 / DGX-2 and later) speedup Detection 3x, Training 2x
• OPENCV=1 to build with OpenCV 3.x/2.4.x - allows to detect on video files and video streams from network cameras or web-cams
• DEBUG=1 to bould debug version of Yolo
• OPENMP=1 to build with OpenMP support to accelerate Yolo by using multi-core CPU
• LIBSO=1 to build a library darknet.so and binary runable file uselib that uses this library. Or you can try to run so LD_LIBRARY_PATH=./:$LD_LIBRARY_PATH ./uselib test.mp4 How to use this SO-library from your own code - you can look at C++ example: https://github.com/AlexeyAB/darknet/blob/master/src/yolo_console_dll.cpp or use in such a way: LD_LIBRARY_PATH=./:$LD_LIBRARY_PATH ./uselib data/coco.names cfg/yolov3.cfg yolov3.weights test.mp4

3. 训练经验

• 首先对数据集进行检错，使用提供的以下库进行检测：

https://github.com/AlexeyAB/Yolo_mark

• 何时中止训练

  • avg loss再也不降低的时候

  • 一般每一个类须要2000-4000次迭代训练便可

  • 防止过拟合：须要在Early stopping point中止训练

    使用如下命令：

    darknet.exe detector map...

    建议训练的时候带上-map,能够画图

4. 提高检测效果

• random=1能够设置适应多分辨率

• 提高分辨率：416--> 608等必须是32倍数

• 从新计算你的数据集的anchor:(注意设置的时候计算问题)

  darknet.exe detector calc_anchors data/obj.data -num_of_clusters 9 -width 416 -height 416

• 检查数据集经过https://github.com/AlexeyAB/Yolo_mark

• 数据集最好每一个类有2000张图片，至少须要迭代2000*类的个数

• 数据集最好有没有标注的对象，即负样本，对应空的txt文件，最好有多少样本就设计多少负样本。

• 对于一张图有不少个样本的状况，使用max=200属性(yolo层或者region层)

• for training for small objects - set layers = -1, 11 instead of https://github.com/AlexeyAB/darknet/blob/6390a5a2ab61a0bdf6f1a9a6b4a739c16b36e0d7/cfg/yolov3.cfg#L720 and set stride=4 instead of https://github.com/AlexeyAB/darknet/blob/6390a5a2ab61a0bdf6f1a9a6b4a739c16b36e0d7/cfg/yolov3.cfg#L717

• 训练数据须要知足如下条件：

  • train_network_width * train_obj_width / train_image_width ~= detection_network_width * detection_obj_width / detection_image_width
  • train_network_height * train_obj_height / train_image_height ~= detection_network_height * detection_obj_height / detection_image_height

• 为了加速训练，能够作fine-tuning而不是从头开始训练，设置stopbackward=1在网络的结束部分（以####做为分割）

• 在训练完之后，进行目标检测的时候，能够提升网络的分辨率，以便恰好检测小目标。

  • 不须要从新训练，须要使用原先低分辨率的权重，测用更高分辨率。
  • 为了获得更高的检测效果，能够提高分辨率至608*608甚至832*832

5. 总结

为了小目标：

• 提高分辨率
• 在测试时候提高分辨率
• 数据集添加跟正样本数量同样多的负样本
• 数据集每一个类至少2000张，训练迭代次数2000*classes个数
• 设置本身数据集的anchor

6. AlexeyAB大神改进

• web-cam版本：

./darknet detector demo ... -json_port 8070 -mjpeg_port 8090

• 计算mAP, F1, IoU, Precision-Recall

./darknet detector map ...

• 展现map-loss曲线(须要opencv)

./darknet detector train cfg/voc.data cfg/yolo.cfg -dont_show -mjpeg_port 8090 -map

• 计算聚类产生的anchor

./darknet detector calc_anchors data/voc.data -num_of_clusters 12 -width 608 -height 608

• 分离前部基础网络

./darknet partial cfg/darknet19_448.cfg darknet19_448.weights darknet19_448.conv.23 23

• 测试opencv

./darknet imtest data/eagle.jpg

• 阈值设置

-thresh 0