UltraLight-VM-UNet
UltraLight-VM-UNet copied to clipboard
[arXiv] The official code for "UltraLight VM-UNet: Parallel Vision Mamba Significantly Reduces Parameters for Skin Lesion Segmentation".
作者你好,我认为你的工作非常令人振奋,麻烦请问一下,我看代码里面并没有加载mamba的预训练,如果使用的话,结果会不会更好呢
Reading PH2 ['/home/chen/datasets/PH2/images/IMD226.png', '/home/chen/datasets/PH2/images/IMD156.png', '/home/chen/datasets/PH2/images/IMD384.png', '/home/chen/datasets/PH2/images/IMD048.png', '/home/chen/datasets/PH2/images/IMD133.png', '/home/chen/datasets/PH2/images/IMD436.png', '/home/chen/datasets/PH2/images/IMD360.png', '/home/chen/datasets/PH2/images/IMD390.png', '/home/chen/datasets/PH2/images/IMD331.png', '/home/chen/datasets/PH2/images/IMD144.png', '/home/chen/datasets/PH2/images/IMD063.png', '/home/chen/datasets/PH2/images/IMD423.png', '/home/chen/datasets/PH2/images/IMD152.png', '/home/chen/datasets/PH2/images/IMD365.png', '/home/chen/datasets/PH2/images/IMD389.png', '/home/chen/datasets/PH2/images/IMD356.png', '/home/chen/datasets/PH2/images/IMD041.png', '/home/chen/datasets/PH2/images/IMD198.png', '/home/chen/datasets/PH2/images/IMD427.png', '/home/chen/datasets/PH2/images/IMD418.png', '/home/chen/datasets/PH2/images/IMD150.png', '/home/chen/datasets/PH2/images/IMD058.png', '/home/chen/datasets/PH2/images/IMD061.png', '/home/chen/datasets/PH2/images/IMD020.png', '/home/chen/datasets/PH2/images/IMD008.png', '/home/chen/datasets/PH2/images/IMD371.png', '/home/chen/datasets/PH2/images/IMD169.png', '/home/chen/datasets/PH2/images/IMD240.png',...
Hi in your ablation experiments where you have used Conv in encoder and decoder, isn‘t that same as UNET? why then do you have for the Conv variants so less...
How important are SAB and CAB? I think in some other papers people just used simple bridge connection , which seems to work fine. Not using SAB anf CAB would...
训练特别慢
我在3090GPU上训练512x512的分割图片,训练速度巨慢,想问下,这个合理嘛?
Hello, this version of the package has been deprecated, what version are you using?
可以用于多分类吗
VM_Unet有用于二分类和多分类的图片的不同训练代码,请问UItraLight-VM-Unet有吗
ImportError: /public/home/ncu_418000230030/anaconda3/envs/Uvmunet/lib/python3.8/site-packages/causal_conv1d_cuda.cpython-38-x86_64-linux-gnu.so: undefined symbol: _ZN3c104cuda20CUDACachingAllocator9allocatorE 请问这个问题怎么解决
请问这个文件在哪里。。
Good work! But it seems that in the model you proposed, the PVMlayer directly uses Albert Gu's Mamba instead of the SS2D in Vmamba. However, in Figure 4 (b) of...