Closed LeeX-Bruce closed 3 months ago
Hi, LeeX-Bruce
We are grateful that you points out bugs in the datasets.py
file. We directly employ the dataloader files from the repository https://github.com/ZJLAB-AMMI/Q-Net/tree/main/dataloaders.
We will update the dataloader code files according to your comments
Okay, thanks for the reply.
Hello author. I found a bug in the
datasets.py
file with settings that don't match the description in the paper.The
__getitem__
method inclass TrainDataset
, which should get the index of the slices of this patient that contain theexclude_label
in caseexclude_label
is passed in, has the following original code:Its does not achieve the above effect. If my
exclude_label
is [1, 2, 3, 4], then theexclude_idx
I get in patient 38 and thesli_idx
I get later is shown below:Here is the code I corrected(Probably?):
The result is as follows: