vs网站建设弹出窗口代码c,电子商务网站建设作业代码,刷评论网站推广,网易博客 wordpressYOLOv8创新魔改教程#xff08;一#xff09;如何进行模块创新
YOLOv8创新魔改教程
本人研一#xff0c;最近好多朋友问我要如何修改模型创新模块#xff0c;就想着不如直接开个专栏歇一歇文章#xff0c;也算是对自己学习的总结#xff0c;本专栏以YOLOv8为例#xf…YOLOv8创新魔改教程一如何进行模块创新
YOLOv8创新魔改教程
本人研一最近好多朋友问我要如何修改模型创新模块就想着不如直接开个专栏歇一歇文章也算是对自己学习的总结本专栏以YOLOv8为例后续会出其他模型的手把手叫大家如何从零创新模块喜欢的朋友可以点个关注喽。
一寻找创新点
我们想对模型创新时要首先寻找创新点如何寻找对于研究目标是图像的朋友可以关注一下CVPR ICCV这些计算机视觉的会议大多情况下我们的创新点就来自于这些期刊。
二github找创新点源码
这里我以《SCConv: Spatial and Channel Reconstruction Convolution for Feature Redundancy》这篇文章为例。大家可以看到这篇文章提出了一个叫SCConv的卷积模块。 现在大家就可以去github找创新点源码了有的文章会提供有的则不会不过也没事github总有人会发表出来大家直接去搜就可以。https://github.com/cheng-haha/ScConv/blob/main/ScConv.py
三将代码整合到自己的网络中
到这一步有的同学可能会开始头疼我代码能力不行我不会编程等等等等我想和大家说的是这一步超级的简单的只需要会复制粘贴就好了
代码一般整合的部位
在YOLOv8中我们的代码可以添加到 ultralytics/nn/modules/block.py中接下来我带大家走一遍流程。
如何添加
第一步打开 ultralytics/nn/modules/block.py这个文件直接将我们找到的代码复制进来。 第二步将创新模块导入大家要把每个class的名称导入到block的__all__ 的括号内ultralytics/nn/modules/init.py文件中from .block import的括号内和__all__ (的括号内一共是三处大家不要漏掉。 进行融合
使用YOLOv8的同学可能会发现发表的论文大多数是对c2f模块的创新所以我也举个例子方便大家参考。大家在block.py中找到c2f模块可以看到里面有四处卷积可以供我们修改其中Bottleneck有两处。 大家在修改的时候建议大家开个新模块改个名字在修改比如要修改c2f
三测试代码
修改yaml文件在这里建议大家创建一个新的。 将c2f换成我们融合后的当然也可以把卷积换成新的卷积。 运行成功可能会出现一些报错无非是模块没添加成功大家根据错误提示修改就可以。 感谢各位的点赞与支持下一篇会更新添加注意力机制相关内容。