rotated_retinanet_obb_vmamba_base_fpn_1x_dota_le90.py

_base_ = './rotated_retinanet_obb_vmamba_tiny_fpn_1x_dota_le90.py'

pretrained = 'data/pretrained/vssm_base_0229_ckpt_epoch_237.pth'

model = dict(
    backbone=dict(
        # copied from https://github.com/MzeroMiko/VMamba/blob/main/detection/configs/vssm1/mask_rcnn_vssm_fpn_coco_base.py
        dims=128,
        depths=(2, 2, 15, 2),
        ssm_ratio=2.0,
        drop_path_rate=0.6,
        use_checkpoint=True,
        pretrained=pretrained),
    # neck=dict(in_channels=[128, 256, 512, 1024]))
    neck=dict(in_channels=[256, 512, 1024]))

fp16 = None