【实验】ConvNeXt+UperNet

cccc楚染rrrr

已于 2023-02-18 16:06:47 修改

阅读量1.4k

点赞数

分类专栏：实验文章标签： python 人工智能深度学习计算机视觉

于 2023-02-18 16:01:00 首次发布

本文链接：https://blog.csdn.net/qq_43606119/article/details/129100913

版权

我的需求：复现convnext网络模型（convnext+upernet），部署到自己的框架代码中，用于语义分割任务（vaihingen数据集）。

前提：根据mmsegmentation框架中的手册安装mmseg、mmcv等库。

进入convnext源码网站：https://github.com/facebookresearch/ConvNeXt
进入semantic_segmantation/文件下
convnext代码在backbone/convnext.py中，
其中需要修改的地方：
（1）我的代码中from mmcv_custom import load_checkpoint总是报错mmcv_custom模块找不到，因此将此段代码注释掉，换成from mmcv.runner import _load_checkpoint，这里是参考了mmsegmentation中poolformer代码的处理方法。
（2）修改代码中使用了load_checkpoint的部分（在ConvNeXt类中）

    def init_weights(self, pretrained=None):
        """Initialize the weights in backbone.
        Args:
            pretrained (str, optional): Path to pre-trained weights.
                Defaults to None.
        """

        def _init_weights(m):
            if isinstance(m, nn.Linear):
                trunc_normal_(m.weight, std=.02)
                if isinstance(m, nn.Linear) and m.bias is