[修改记录]对于yolov5的head部分里的一些模块进行了修改替换

爱改网络的法式小面包

已于 2023-09-11 10:23:11 修改

阅读量1.3k

点赞数 2

分类专栏： yolov5系列改进思路文章标签： YOLO python 深度学习

于 2023-03-03 21:18:02 首次发布

本文链接：https://blog.csdn.net/PLH19990227/article/details/129326553

版权

yolov5系列改进思路专栏收录该内容

6 篇文章 7 订阅 ¥19.90 ¥99.00

订阅专栏

超级会员免费看

本文介绍了对YOLOv5头部结构的改进，将Conv替换为GSConv，并用VoVGSCSP替换C3。在model/common.py中添加了相关函数，在yolo.py和yolov5s.yaml中进行相应修改。初步测试显示识别精度和mAP有所提升，但因数据集限制，尚不清楚在大规模数据和更多迭代次数下效果如何。

摘要由CSDN通过智能技术生成

第一步修改是将head部分里面的Conv替换成了GSConv，GSConv函数是基于Conv函数修改的，主要的修改内容如下：

1.在model/common.py里添加GSConv的函数代码，添加的代码段如下：

#---------------自己添加的GSConv------------
class GSConv(nn.Module):
    def __init__(self,c1,c2,k=1,s=1,g=1, act=True):
        c_ = c2 // 2
        self.cv1 = Conv(c1, c_, k, s, None, g, act=act)
        self.cv2 = Conv(c_, c_, 5, 1, None, c_, act=act)

    def forward(self,x):
        x1 = self.cv1(x)
        x2 = torch.cat((x1,self.cv2(x1)),1)
        # shuffle
        b, n, h, w = x2.data.size()
        b_n = b * n  // 2
        y = x2.reshape(b_n, 2, h * w)
        y = y.permute(1, 0, 2)
        y = y.reshape(2, -1, n // 2, h, w)

了解本专栏