search

lofrienger / S2ME

S2ME: Spatial-Spectral Mutual Teaching and Ensemble Learning for Scribble-supervised Polyp Segmentation (MICCAI 2023)
15 stars 1 forks source link

Hello #5

Open yike5022 opened 3 months ago

yike5022 commented 3 months ago

Hello, I want to use your model to run the test set directly to see the prediction results, but I can't find the test file, how can I test it?

lofrienger commented 3 months ago

Hi, you can use the checkpoint in the ./model folder of this repo to test.

yike5022 commented 3 months ago

Hello, which file do I need to use?,Is it ./code/train_s2me.py?

lofrienger commented 3 months ago

Yes, you can refer to lines 253-268 for testing.

yike5022 commented 3 months ago

Hello, I see that there is only one pth file under ./model, but I need to load the parameters of two models when testing, what should I do to load this model?

lofrienger commented 3 months ago

Hi, the provided checkpoint is for the UNet branch, where our reported results come from. During training, we monitored both branches. But during testing, you don't have to use the YNet model.

yike5022 commented 3 months ago

I fully understand what you mean, thank you for your reply