search

shenqq377 / Q-Net

Python codes to implement Q-Net, a meta-learning method for few shot medical image segmentation
https://arxiv.org/abs/2208.11451
42 stars 5 forks source link

I would like to ask how to experiment with setup 2 #4

Closed WangBingJian233 closed 1 year ago

WangBingJian233 commented 1 year ago

Hello author, thank you very much for your outstanding contribution. I would like to ask how to experiment with setup 2 (i.e. guarantee that the test class is truly "invisible" to the model)?

shenqq377 commented 1 year ago

Hi, please check the class TrainDataset in datasets.py. In the bash scripts or config.py, you could find the variable "exclude_label" that defines the test label index to be excluded. In the class TrainDataset, you could find that we exclude slices that contain "exclude_label" [Line 226-233].