search

junjie-shentu / CXR-IRGen

Implementation of the paper "CXR-IRGen: An Integrated Vision and Language Model for the Generation of Clinically Accurate Chest X-Ray Image-Report Pairs" (WACV 2024)
13 stars 2 forks source link

question for the dataset #3

Open Andyfever123 opened 5 months ago

Andyfever123 commented 5 months ago

Hello, if I want to use IU-xray or another dataset to carry out the experiments in your paper, what parameters and files should I modify? Looking forward to receiving your reply, thank you very much!

junjie-shentu commented 4 months ago

Hi Andy! Sorry for the late reply! If you want to use other datasets, it is recommended to check the dataset files diffusion_model_dataset.py, lm_dataset.py and prior_dataset.py in the dataset folder. Because we applied some filtering to the dataset in our code, you might need to remove them to adapt the code to other datasets. You can download our sample data files via link https://drive.google.com/file/d/1hQydkKn4uAKbh781AlKjNMeqmTHE1LQ7/view?usp=drive_link, and construct your own data files.