AkitsukiM / VMamba-DOTA

25 stars 0 forks source link

VMamba做分割 #2

Open DuHao55 opened 6 months ago

DuHao55 commented 6 months ago

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

AkitsukiM commented 5 months ago

没什么区别。以及我刚刚基于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)

推荐使用!

DuHao55 commented 5 months ago

好的!谢谢~