将 CascadetabNet 代码从 mmdet 2 调整为 mmdet 3

Question

我正在尝试将 CascadetabNet 代码改编为 pytorch 2.0。 显然这个mmdet 3是需要的，所以我用mim安装了它

mim install "mmdet>=3.0.0rc7" 

所以代码看起来像这样

from mmdet.apis import init_detector, inference_detector
import mmcv
# Load model
config_file = 'path/to/config.py'
checkpoint_file = '/content/epoch_36.pth'
model = init_detector(config_file, checkpoint_file, device='cuda:0')

配置文件的开头有这个预训练参数

# model settings
model = dict(
    type='CascadeRCNN',
    # num_stages=3,
    pretrained='open-mmlab://msra/hrnetv2_w32',
    backbone=dict(

mmdet 3.x 显然不存在

TypeError: __init__() got an unexpected keyword argument 'pretrained'

有人知道如何适应这个吗？

我尝试删除参数，它给了我这个

AttributeError: 'NoneType' object has no attribute 'rpn'

Answer 1

我也有同样的问题。你找到解决办法了吗？