关于Yolov7-tiny模型瘦身（param、FLOPs）碎碎念

这篇具有很好参考价值的文章主要介绍了关于Yolov7-tiny模型瘦身（param、FLOPs）碎碎念。希望对大家有所帮助。如果存在错误或未考虑完全的地方，请大家不吝赐教，您也可以点击"举报违法"按钮提交疑问。

优于模型落地需要，yolov7-tiny版本的参数量(Param)为6.2M，FLOPs在输入为320*320下为3.46G，想着还能不能再进行减少这两者或其中之一(精度方面还未做效果验证)。

于是翻遍了github，没有关于v7的剪枝方法，替换backbone也大部分是基于yolov5s的，眼花缭乱中看到yolov7的github官方作者的一句回复：

关于Yolov7-tiny模型瘦身（param、FLOPs）碎碎念

接着去看了一下什么是ELAN，相关细节知识不做复制粘贴了，可以去看各大神的代价解读。在yolov7-tiny.yaml中，就是这个在被stride为2的卷积下采样后，被Concat整合前的这四个连续卷积层。

关于Yolov7-tiny模型瘦身（param、FLOPs）碎碎念

这种结构在标准版和tiny版的backbone和head中都有，只是tiny中少了两个卷积层的堆叠，激活函数变成了LeakyReLU，每个branch的操作中，输入通道都是和输出通道保持一致，接下来就是小白改结构碰的壁了。

每层网络都有四个参数，[from, number, module, args]，以Conv层为例，第一个from表示本层的输入来自哪一层的输出，-1表示上一层，-2表示上上层，以此类推，若是正数n则表示从头数的第n层；number表示本层堆叠几次；module表示该层的种类，在models.common中都能找到class；args则是相关参数，如Conv中是通道数、卷积核尺寸、步长、激活函数等。

当时由于不懂from的含义，就硬删ELAN中的某层卷积（from为-2的卷积不要删），然后train.py，结果可想而知。

关于Yolov7-tiny模型瘦身（param、FLOPs）碎碎念

根据log找到是构建网络时报的错，然后继续研究。。。因为网络各层是堆叠的，删除某层对后面的层数当然会有影响，所以from不为-1的值肯定要改，经过一系列小学数学的操作后，层和from对应上了，还是报错。。。接着研究，在yolov7.yaml和v7-tiny.yaml中发现Concat层的from是[-1,-2,-3,-4,-5,-6]和[-1,-2,-3,-4]的区别，所以当from是数组时，表示上一层，上2层，上3层..以此类推。所以ELAN中删一层，[-1,-2,-3,-4]也要变成[-1,-2,-3]，最后一层IDetect也需要修改，该层的from表示第[n,n+1,n+2]层的输出作为Detect模块的输入，接着run，成功！

关于Yolov7-tiny模型瘦身（param、FLOPs）碎碎念