HorNet+YOLOv5改进方案

Dropout~~

已于 2024-03-05 16:43:16 修改

阅读量3.7k

点赞数 5

CC 4.0 BY-SA版权

分类专栏： YOLOv5改进涨点专栏文章标签：计算机视觉 python 深度学习

于 2022-11-06 20:08:02 首次发布

本文链接：https://blog.youkuaiyun.com/ZzzzzKnight/article/details/127717177

YOLOv5改进涨点专栏专栏收录该内容

26 篇文章 ¥59.90 ¥99.00

订阅专栏

本文介绍了如何将HorNet模块应用于YOLOv5的改进过程，包括在common.py文件中添加代码，创建yolov5GNCSP.yaml配置文件，以及在YOLO.py的parse_model函数中进行修改的详细步骤。

HorNet理论请参考论文：https://arxiv.org/pdf/2207.14284.pdf

🚀yolov5加HorNet模块代码

1.在 common.py 文件中添加如下代码

部分代码如下

class Gnconv(nn.Module):
    def __init__(self, dim, order=5, gflayer=None, h=14, w=8, s=1.0):
        super().__init__()
        self.order = order
        self.dims = [dim // 2 ** i for i in range(order)]
        self.dims.reverse()
        self.proj_in = nn.Conv2d(dim, 2*dim, 1)
 
        if gflayer is None:
            self.dwconv = get_dwconv(sum(self.dims), 7, True)
        else:
            self.dwconv = gflayer(sum(self.dims), h=h, w=w)
        
        s

了解本专栏