How to Run Code for IU Xray Dataset

deng1604 commented 2 weeks ago

在dataset.py中，IU Xray数据集和 MIMIC-CXR 数据集的定义是不一样的

class IuxrayMultiImageDataset(BaseDataset):
    ......

    sample = (image_id, image, report_ids, report_masks, seq_length)
    return sample

class MimiccxrSingleImageDataset(BaseDataset):
    ......
    sample = (image_id, image, report_ids, report_masks, label, seq_length)

    return sample

而在dataloaders.py中，

def collate_fn(data):
    images_id, images, reports_ids, reports_masks, label, seq_lengths = zip(*data)
    .......

这里只对应MIMIC-CXR数据集的，那请问作者，IU Xray数据集，你是怎样进行训练得到结果的呢？

xzp9999 commented 2 weeks ago

IU Xray用的另外一套代码，后边抽空放上来

deng1604 commented 1 week ago

好的，谢谢作者🌹

xzp9999 / GMoD-mian

How to Run Code for IU Xray Dataset #2