2403_82630476的博客

原创 YOLOv5改进 | 引入解耦头DecoupledHead（实测涨点）

在models/common.py的任意位置添加下面DecoupledHead代码即可 （我一般在common.py最下方添加，比较方便）最后一步~~ 修改train.py 的 --cfg 将默认的yolov5s.yaml修改为我们之前新建的。在models的yolo.py中的两个位置将默认的yolov5s.yaml修改成我们新建的。如图所示我们添加解耦头成功！（2） 在Model类里的图中位置添加下图红框的代码。然后运行train.py就可以运行啦！我们在修改网络模型时只更改最后一层的。