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

改进YOLO系列 | 添加轻量化Decouple_Head 和 ASFF_Head_第1张图片


改进YOLO系列 | 添加轻量化Decouple_Head 和 ASFF_Head_第2张图片

绿色为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_HeadYOLOX的头结构几乎相同,但这次的添加方式和我蓝皮书介绍的那篇是不同的,而且这个头优化的比较好,所以参数量和计算量都下降了,效果有可能和YOLOX的头有差距,。

你可能感兴趣的:(YOLOv5/v7进阶实战,YOLO,python,numpy)