JCruan519 / VM-UNet

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

I have changed images dand masks which in my data into .h5 and .npz #82

Open kukunanhaixiao opened 1 month ago

kukunanhaixiao commented 1 month ago

im also successfully ran your train.py , exactly i got best.pth now there are two problems. first is i want get .png files after running some ppredict.py file. but i didn't find your predict.py . another problem is i got .nii file while running the train_spayse.py fil, are they the predicted result? i want get 2-d imas result, how can i modify your program? thank you for your answer!