JCruan519/VM-UNet

inference code

Opened this issue · 3 comments

tubixiansheng commented

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

JCruan519 commented

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

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

JCruan519 commented

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