本文介绍
在做一些比较实验中，特别是在模型轻量化时,如何更精确的查看和对比我们的改进模块时候有效，是否有提升呢？
这时候我们就可以打印改进模型中每一层的耗时和GFLOPS来比较不同模块的占用量。这样就可以有针对性的改进我们的模型，并且在写论文中直接展示某一层的改进效果，会更加的直观，也能够丰富论文内容。
在下方的实现效果中，可以看出来能够打印每一层的相关耗时time，计算量GFLOPs，参数量params等，可查看改进后的任意模型，并且可调节batch-size。
专栏目录：YOLOv9改进目录一览 | 涉及卷积层、轻量化、注意力、损失函数、Backbone、SPPF、Neck、检测头等全方位改进
专栏地址：
订阅专栏 解锁全文
扫一扫
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
