search

ge-xing / SegMamba

SegMamba: Long-range Sequential Modeling Mamba For 3D Medical Image Segmentation
https://arxiv.org/abs/2401.13560
290 stars 13 forks source link

2D data #18

Open alex-stenger opened 2 months ago

alex-stenger commented 2 months ago

Hi, I would like to know how to configure the SegMamba class to input 2D data inside (typically of form (B,C,H,W), where the model as is it takes data of the form (B,C,Z,H,W)). Thanks a lot !

ge-xing commented 2 months ago

Sorry. This model only receives 3D medical inputs. Many designs in this model are all for the 3D data.