search

ziyangwang007 / Mamba-UNet

Mamba-UNet Zoo
Apache License 2.0
425 stars 29 forks source link

test.py #5

Closed zhangfeifei1-ab closed 6 months ago

zhangfeifei1-ab commented 7 months ago

Hello, when I was testing your code, I encountered the following issue: In net_factory.py, the Mamba-Unet model is not loaded, causing the test to fail to load the net.

ziyangwang007 commented 6 months ago

hi, try to load model 'from networks.vision_mamba import MambaUnet as ViM_seg'. hope be helpful.