├── .gitignore ├── LICENSE ├── README.md ├── configs ├── config_setting.py ├── config_setting_synapse.py └── config_setting_v2.py ├── data └── README.md ├── datasets └── dataset.py ├── engine.py ├── engine_synapse.py ├── models ├── unet_v2 │ ├── UNet_v2.py │ ├── __init__.py │ └── pvtv2.py ├── vmunet │ ├── vmamba.py │ ├── vmunet.py │ └── vmunet_v2.py └── vmunet_old │ ├── vmamba.py │ └── vmunet.py ├── pre_trained_weights └── README.md ├── train.py ├── train_gim.py ├── train_isic_all.py ├── train_npy.py ├── train_synapse.py ├── unet_v2 ├── UNet_v2.py ├── __init__.py └── pvtv2.py └── utils.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nobodyplayer1/VM-UNetV2/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nobodyplayer1/VM-UNetV2/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nobodyplayer1/VM-UNetV2/HEAD/README.md -------------------------------------------------------------------------------- /configs/config_setting.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nobodyplayer1/VM-UNetV2/HEAD/configs/config_setting.py -------------------------------------------------------------------------------- /configs/config_setting_synapse.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nobodyplayer1/VM-UNetV2/HEAD/configs/config_setting_synapse.py -------------------------------------------------------------------------------- /configs/config_setting_v2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nobodyplayer1/VM-UNetV2/HEAD/configs/config_setting_v2.py -------------------------------------------------------------------------------- /data/README.md: -------------------------------------------------------------------------------- 1 | You could put the datasets here. -------------------------------------------------------------------------------- /datasets/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nobodyplayer1/VM-UNetV2/HEAD/datasets/dataset.py -------------------------------------------------------------------------------- /engine.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nobodyplayer1/VM-UNetV2/HEAD/engine.py -------------------------------------------------------------------------------- /engine_synapse.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nobodyplayer1/VM-UNetV2/HEAD/engine_synapse.py -------------------------------------------------------------------------------- /models/unet_v2/UNet_v2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nobodyplayer1/VM-UNetV2/HEAD/models/unet_v2/UNet_v2.py -------------------------------------------------------------------------------- /models/unet_v2/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/unet_v2/pvtv2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nobodyplayer1/VM-UNetV2/HEAD/models/unet_v2/pvtv2.py -------------------------------------------------------------------------------- /models/vmunet/vmamba.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nobodyplayer1/VM-UNetV2/HEAD/models/vmunet/vmamba.py -------------------------------------------------------------------------------- /models/vmunet/vmunet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nobodyplayer1/VM-UNetV2/HEAD/models/vmunet/vmunet.py -------------------------------------------------------------------------------- /models/vmunet/vmunet_v2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nobodyplayer1/VM-UNetV2/HEAD/models/vmunet/vmunet_v2.py -------------------------------------------------------------------------------- /models/vmunet_old/vmamba.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nobodyplayer1/VM-UNetV2/HEAD/models/vmunet_old/vmamba.py -------------------------------------------------------------------------------- /models/vmunet_old/vmunet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nobodyplayer1/VM-UNetV2/HEAD/models/vmunet_old/vmunet.py -------------------------------------------------------------------------------- /pre_trained_weights/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nobodyplayer1/VM-UNetV2/HEAD/pre_trained_weights/README.md -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nobodyplayer1/VM-UNetV2/HEAD/train.py -------------------------------------------------------------------------------- /train_gim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nobodyplayer1/VM-UNetV2/HEAD/train_gim.py -------------------------------------------------------------------------------- /train_isic_all.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nobodyplayer1/VM-UNetV2/HEAD/train_isic_all.py -------------------------------------------------------------------------------- /train_npy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nobodyplayer1/VM-UNetV2/HEAD/train_npy.py -------------------------------------------------------------------------------- /train_synapse.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nobodyplayer1/VM-UNetV2/HEAD/train_synapse.py -------------------------------------------------------------------------------- /unet_v2/UNet_v2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nobodyplayer1/VM-UNetV2/HEAD/unet_v2/UNet_v2.py -------------------------------------------------------------------------------- /unet_v2/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /unet_v2/pvtv2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nobodyplayer1/VM-UNetV2/HEAD/unet_v2/pvtv2.py -------------------------------------------------------------------------------- /utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/nobodyplayer1/VM-UNetV2/HEAD/utils.py --------------------------------------------------------------------------------