ConvNeXt 的 不同版本实现 推理代码 的参考配置

发布时间：2026/7/5 14:28:12

ConvNeXt 的 不同版本实现 推理代码 的参考配置flyfish推理时不同的版本要注意区别为了简单crop_size224, resize_size256需要按照不同的配置进行编写例如crop_size224, resize_size230crop_size224, resize_size232维度facebookresearch 官方原版torchvision 官方实现特征层组织用nn.ModuleList拆分管理downsample_layers4个Stem 3次下采样stages4个特征提取阶段forward 中循环遍历先下采样再经过阶段全部特征层封装进一个大nn.Sequential命名为features按线性顺序排列Stem → Stage1 → 下采样1 → Stage2 → 下采样2 → Stage3 → 下采样3 → Stage4分类头结构全局平均池化 → 最终 LayerNorm → Linear 分类头对应属性self.normself.head自适应平均池化 →classifier序列对应结构LayerNorm2d Flatten Linear典型权重键名downsample_layers.0.0.weightstages.0.0.dwconv.weightnorm.weighthead.weightfeatures.0.0.weightfeatures.1.0.block.0.weightclassifier.0.weightclassifier.2.weight1K 类别数 100022K 类别数 21841