目标检测:记录mmdetection替换backbone为ResNeSt

本文档详细介绍了如何在mmdetection框架中替换backbone为ResNeSt,包括定义ResNeSt模块,导入相关组件,适配ResNeSt和ResNet文件,以及在配置文件中使用新的backbone。
摘要由CSDN通过智能技术生成

提示:文章写完后,目录可以自动生成,如何生成可参考右边的帮助文档


前言

实验需要将现有开源目标检测代码的backbone替换为ResNeSt,开源代码基于mmdetection v1开发,mmcv版本较低(0.2.16),随着版本更新, 目前mmdetection v2.7支持ResNeSt。由于版本更新较大,新旧版本的mmdetection代码架构及实现差别较大,仅仅按照官网教程已不能将旧版mmdetection的backbone替换为ResNeSt,需改动多处,本文记录低版本mmdetection的backbone的替换过程。

提示:以下是本篇文章正文内容,下面案例可供参考

一、基本环境配置

ubuntu18.04 python3.7 pytorch1.7.1 CUDA11.1 mmcv0.5.9

二、实现过程

整体步骤按照mmdetection官网教程
mmdetection官网提供的自定义模型教程

1.Define resnest

将resnest.py和resnet.py粘贴至开源代码(基于旧版mmdetection开发)

在新版mmdetection (v2.10)找到resnest.py(mmdet/models/backbones/resnest.py)
先直接粘贴至开源代码中(mmdet/models/backbones/resnest.py)

由于resnest.py依赖resnest.py中的一些package和函数,因此也需要将resnet.py替换掉。方法同上

2. Import the module

在mmdet/models/backbones/init.py添加resnest

from .resnet import ResNet
from .resnext import ResNeXt
from .ssd_vgg import SSDVGG
from .hrnet import HRNet
from .resnest import ResNeSt


__all__ = ['ResNet','ResNeSt','ResNeXt', 'SSDVGG', 
  • 2
    点赞
  • 42
    收藏
    觉得还不错? 一键收藏
  • 8
    评论
评论 8
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值