当前位置:   article > 正文

yoloV5 增加小目标检测层_yolov5增加检测头

yolov5增加检测头

yolo

yoloV5 6.0增加小目标检测

原始网络结构:

image-20230529152226935

原始网络有三个检测头: 分别是 80x80 (小目标) 40x40(中目标) 160x160(小目标),现在要增加小目标检测层,就可以在80 x 80 的上一步,也就是 160 x 160 尺寸增加

image-20230529154406204

为了检测160 x 160的检测头,因此需将backbon中160 与NECK中的 160 相加(残差网络),但原始网络中 NECK没有60 x 160的特征图,但可以对 80x80的特征图再进行一次上采样,得到160x160的特征图,并于backbon中60x160的特征图连接即可得到下图的网络图

image-20230529155101831

取消红框,增减蓝框区域

取消红框的原因:特征金字塔变长了,红框右侧原来是最底层,因为直接从右侧获取特征图即可,增加后不是最后一层了,需要从它的下侧获取特征图(可参考 40x40 20x20获取特征图的方式)

逻辑:NECK中 80的特征图经过上采样变成160,可与backbon中160的特征图相加,得到一个160的特征图,然后经过下采样,又获得80的特征图(特诊金字塔 FPN + TPN)并经过C3传给检测头

修改yaml

增加小目标层 160 *160并送入检测头

# YOLOv5 
声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/很楠不爱3/article/detail/77642
推荐阅读
相关标签