oriented_rcnn_vheat_small_fpn_1x_dota_le90_ms.py

_base_ = './oriented_rcnn_vheat_tiny_fpn_1x_dota_le90_ms.py'

pretrained = 'data/pretrained/vHeat_small.pth'

model = dict(
    backbone=dict(
        # copied from https://github.com/MzeroMiko/vHeat/blob/main/detection/configs/vheat/mask_rcnn_fpn_coco_small.py
        post_norm=True,
        depths=(2, 2, 18, 2),
        dims=96,
        drop_path_rate=0.3,
        layer_scale=1.e-5,
        img_size=512,
        use_checkpoint=True,
        pretrained=pretrained))

fp16 = None