YOLO 训练和检测技巧

最新推荐文章于 2024-10-04 16:01:05 发布

pangdawa

最新推荐文章于 2024-10-04 16:01:05 发布

阅读量3.2k

点赞数 4

分类专栏：深度学习文章标签： YOLO 训练检测精度

深度学习专栏收录该内容

11 篇文章 0 订阅

订阅专栏

注意根据特定任务对 anchor 进行修改，方法参考
https://github.com/pjreddie/darknet/issues/911
./darknet detector calc_anchors data/hand.data -num_of_clusters 9 -width 704 -height 576 -show

https://github.com/AlexeyAB/darknet#how-to-train-to-detect-your-custom-objects
这个文档给出了一些提高精度的方法

提高物体检测性能的方法：
训练前

1 cfg 文件中设置 random = 1

2 提高 cfg中指定的分辨率，比如 height = 608, width = 608 或者更高的32的倍数

3 确保每个样本都有标注并且正确标注

4 对于待检测的物体，训练集中要有相似物体，形状、大小、旋转角度、倾斜、明亮度方面相似
包含物体在不同尺度、旋转、照片、侧面、和背景下的图像有利于训练，每个类别至少2000张不同的图像
至少训练 2000*类别数次

5 提高不包含待检测物体的图像也很重要，就是负样本，这些样本对应的txt为空文件，
负样本的数量最好和正样本数量一致。这条容易被忽略

6 如果单张图像里面需要检测出的物体比较多，那么设置yolo层和region层中 max=200 或者更大值

7 如果是检测小物体（图像resize为416*416 后，小于 16*16 像素），那么在 yolov3.cfg中720 行设置 layers = -1, 11
并且， 717行设置 stride 为4

8 如果训练既有大物体又有小物体的模型，那么使用 yolov3_5l.cfg，这个有 5 yolo层
如果用 tiny模型，那么使用 yolov3-tiny_3l.cfg，
如果用 Spatial-full-model，那么用 yolov3-spp.cfg

9 如果是训练需要区分左右的物体，那么禁用 flip , yolov3cfg 第17行设置 flip = 0

10 一般来说，训练集中物体大小和待检测图像中物体大小有一定约束，大致关系为
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

也就是说，如果训练集中物体宽度占图像的比例要约等于测试集中物体宽度占图像的比例

11 如果要加速训练（可能会降低精度），使用 fine-tuning，而不是迁移学习，设置 548 行 stopbackward=1
执行命令： ./darknet partial cfg/yolov3.cfg yolov3.weights yolov3.conv.81 81 will be created file yolov3.conv.81, then train by using weights file yolov3.conv.81 instead of darknet53.conv.74

12 ？？检测的物体越不同，使用的网络模型就应该越复杂

13 针对cfg 文件中的高宽重新计算 anchor：darknet.exe detector calc_anchors data/obj.data -num_of_clusters 9 -width 416 -height 416
然后设置 yolo层中的 anchor . 需要修改 masks ，是的第一个 yolo层的anchor 大于60*60, 第二个大于 30*30，第三个保持不变
同时要修改 yolo层前面 filters 的个数，公式为