文章目录
摘要
这篇文章主要讲解如何使用InceptionNext主干网络替换YoloV8和YoloV5的主干。
修改InceptionNeXt 模型脚本
首先,要对InceptionNeXt 脚本做修改,去掉Head层,并将返回值改成特征列表。这些特征列表分别是160,80,40,20的特征图。如下图:
直接上代码吧!下面是修改好的代码:
"""
InceptionNeXt implementation, paper: https://arxiv.org/abs/2303.16900
Some code is borrowed from timm: https://github.com/huggingface/pytorch-image-models
"""
from functools import partial
import torch