【YOLOv5】SPP、SPPF模块及添加ASPP模块

最新推荐文章于 2024-11-07 20:19:20 发布

Petra取名废

最新推荐文章于 2024-11-07 20:19:20 发布

阅读量2.3w

点赞数 14

CC 4.0 BY-SA版权

分类专栏：今天报错了吗文章标签：深度学习 cnn python

本文链接：https://blog.youkuaiyun.com/weixin_49221232/article/details/122452513

本文介绍了YOLOv5的配置文件解析，重点探讨了SPP模块，包括其解决CNN全连接层固定输入的问题，以及YOLOv5如何实现SPP。此外，还提及了SPPF和ASPP（空洞空间卷积池化金字塔）的概念，ASPP用于扩大感受野，但在实验中发现效果不明显，提出了对ASPP潜在问题的思考。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

文章目录

YOLOv5配置文件
SPP介绍
YOLOv5中的SPP
SPPF
ASPP

又名，整理整理自己不靠谱的学习记录。

YOLOv5配置文件

YOLOv5通过yaml配置文件来搭建先在common.py文件中构建相同名称的类，再从配置文件中读取到对应的操作后就会对该操作的类实例化。整个网络结构由Model类构成，Model中调用了parse_model()函数，由该函数对配置文件进行解析后调用对应的类进行网络构建，构建后由Model实现后面的处理。
parse_model()函数存在于yolo.py文件中。

def parse_model(d, ch):  # model_dict, input_channels(3)
    logger.info('\n%3s%18s%3s%10s  %-40s%-30s' % ('', 'from', 'n', 'params', 'module', 'arguments'))
    anchors, nc, gd, gw = d['anchors'], d['nc'], d['depth_multiple'], d['width_multiple']
    na = (len(anchors[0]) // 2) if isinstance(anchors, list) else anchors  # number of anchors
    no = na * (nc + 5)  # number of outputs = anchors * (classes + 5)

    layers, save, c2 = [],