UltraLight VM-UNet:一种超轻量级的医学图像分割新网络,提出在并行视觉Mamba中处理特征的方法,参数仅为0.049M,GFLOP为0.060,性能优于LightM-UNet等网络,代码刚刚开源! 点击关注 @CVer官方知乎账号,可以第…
frommodels.UltraLight_VM_UNetimportUltraLight_VM_UNet fromengineimport* importos importsys os.environ["CUDA_VISIBLE_DEVICES"]="0"# "0, 1, 2, 3" fromutilsimport* fromconfigs.config_settingimportsetting_config importwarnings warnings.filterwarnings("ignore") ...
1.The UltraLight VM-UNet has only 0.049M parameters, 0.060 GFLOPs, and a model weight file of only 229.1 KB. 2.Parallel Vision Mamba is a winner for lightweight models. News🚀 (2024.04.04)Added preprocessing step for private datasets. ...
Actions Projects Security Insights Additional navigation options Files main configs dataprepare models results README.md engine.py loader.py test.py train.py utils.py Breadcrumbs UltraLight-VM-UNet / File metadata and controls 83 lines (69 loc) · 2.73 KB ...