AkitsukiM/VMamba-DOTA

VMamba做分割

Opened this issue · 2 comments

我想用Vmaba作为Backbone做分割,请问您注释掉的MMSEG_VSSM相对于MMDET_VSSM,有何改变呢。

没什么区别。以及我刚刚基于VMamba v0.2更新了一个检测代码版本。这一版本更加灵活泛用,可以看到VMamba的作者使用了这样的代码写法兼容检测分割

@MODELS_MMSEG.register_module()
@MODELS_MMDET.register_module()
class MM_VSSM(BaseModule, Backbone_VSSM):
    def __init__(self, *args, **kwargs):
        BaseModule.__init__(self)
        Backbone_VSSM.__init__(self, *args, **kwargs)

推荐使用!

好的!谢谢~