改进YOLO系列 | 添加轻量化Decouple_Head 和 ASFF_Head
Posted 迪菲赫尔曼
tags:
篇首语：本文由小常识网(cha138.com)小编为大家整理，主要介绍了改进YOLO系列 | 添加轻量化Decouple_Head 和 ASFF_Head相关的知识，希望对你有一定的参考价值。
绿色为
ASFF_Head，浅蓝色Decoupled_Head，深蓝色是第三步加的_initialize_dh_biases方法后的效果。
参数量与计算量对比
| 模型 | 参数量 parameters | 计算量GFLOPs |
|---|---|---|
| yolov5s_Head | 7235389 | 16.5 |
| ASFF_Head | 12674847 | 25.0 |
| Decoupled_Head | 8928697 | 22.0 |
结构图
本篇介绍的这个Decouple_Head和YOLOX的头结构几乎相同，但这次的添加方式和我蓝皮书介绍的那篇是不同的，而且这个头优化的比较好，所以参数量和计算量都下降了，效果有可能和YOLOX的头有差距，。
以上是关于改进YOLO系列 | 添加轻量化Decouple_Head 和 ASFF_Head的主要内容，如果未能解决你的问题，请参考以下文章