RyanWangZf / MedCLIP

EMNLP'22 | MedCLIP: Contrastive Learning from Unpaired Medical Images and Texts
394 stars 41 forks source link

Training from scratch with Trainer Class #14

Closed SyedA5688 closed 1 year ago

SyedA5688 commented 1 year ago

Hi there, I am trying to train a MedCLIP model from scratch on the CheXpert and MIMIC-CXR datasets. I cannot seem to find the training script which calls the Trainer class in order to define the model and dataloaders, and train the model from scratch.

Could you by any chance provide the training script which utilizes the Trainer class?

zhaoedf commented 1 year ago

Hi there, I am trying to train a MedCLIP model from scratch on the CheXpert and MIMIC-CXR datasets. I cannot seem to find the training script which calls the Trainer class in order to define the model and dataloaders, and train the model from scratch.

Could you by any chance provide the training script which utilizes the Trainer class?

do you make any progress? i want to train MedCLIP as well. thx.

RyanWangZf commented 1 year ago

Hi, a pretrain example is uploaded under examples/.

Due to the data privacy issue, we can only provide a small piece of sample data under local_data.

Thanks for your attention!

SyedA5688 commented 1 year ago

Thank you!

zhaoedf commented 1 year ago

Hi, a pretrain example is uploaded under examples/.

Due to the data privacy issue, we can only provide a small piece of sample data under local_data.

Thanks for your attention!

Thanks!

Besides,can you provide the data source of COVID? i found some dataset named COVID in kaggle, but the number of samples does not match Table 3. in paper.

What's more, for MIMIC-CXR dataset, you download the original DICOM format file and process it into jpg files? or you just download the jpg from somewhere?