改进YOLO系列 | 添加轻量化Decouple_Head 和 ASFF_Head

Posted 2023-03-05 迪菲赫尔曼

---

绿色为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的头有差距，。