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
517 stars 24 forks source link

inference code #53

Open tubixiansheng opened 6 months ago

tubixiansheng commented 6 months ago

Hello, I'd like to ask if there are any inference codes available?Looking forward to your response.

JCruan519 commented 6 months ago

Hello, the inference code and the training code are merged together. That is, after training, it will automatically select the best weights for inference testing. If you want to use the inference part separately, you can refer to the test_one_epoch function in eggine.py.

hafsa390 commented 6 months ago

Hi @JCruan519 , looks like the validation dataset is used as test dataset during inference for isic.

JCruan519 commented 6 months ago

@hafsa390 Yes, we divide the training and test sets in a 7:3 ratio. This is consistent across all models.