提示:文章写完后,目录可以自动生成,如何生成可参考右边的帮助文档
文章目录
前言
实验需要将现有开源目标检测代码的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官网教程
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',