search

JCruan519 / VM-UNet

(ARXIV24) This is the official code repository for "VM-UNet: Vision Mamba UNet for Medical Image Segmentation".
Apache License 2.0
407 stars 13 forks source link

Error in vmamba.py #4

Open ZhiQiangYang513 opened 5 months ago

ZhiQiangYang513 commented 5 months ago

Hi, I m quite interested in your excellent work and try to use it in my field, but I got a small problem in vmamba.py in the below lines:

an alternative for mamba_ssm (in which causal_conv1d is needed)

try: from selective_scan import selective_scan_fn as selective_scan_fn_v1 from selective_scan import selective_scan_ref as selective_scan_ref_v1 except: pass It looks like that my pycharm can not find the package or function named selective_scan ? and I see the note "an alternative for ...", is that means selecetive_san is not necessary for the whole codes?

JCruan519 commented 4 months ago

@ZhiQiangYang513 Hello, thank you for your interest in our work. Selecetive_san is a key component of Mamba, and it's essential for the entire project. You can install it by using 'pip install mamba_ssm==1.0.1', or you can refer to the installation method directly in the VMamba project (https://github.com/MzeroMiko/VMamba).

Zong-Liang commented 4 months ago

Hello, I have encountered the same problem as you. Have you solved it? I think I have installed mamba_ssm correctly. image image

ZhiQiangYang513 commented 4 months ago

Hello, I have encountered the same problem as you. Have you solved it? I think I have installed mamba_ssm correctly.

image

image

I think it may be ok even this function or package is missing. My code can be run under this situation

tantian616 commented 4 months ago

Hello, I have encountered the same problem as you. Have you solved it? I think I have installed mamba_ssm correctly. image image

I have encountered the same problem as you. Have you solved it?

ZhiQiangYang513 commented 4 months ago

Hello, I have encountered the same problem as you. Have you solved it? I think I have installed mamba_ssm correctly. image image

I have encountered the same problem as you. Have you solved it?

Hi，I ignore this issue， code can be run fine