Hello, I would like to generate an image using clipUViT. What specific parameters do I need?

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

Hello, I would like to generate an image using clipUViT. What specific parameters do I need? #2

Open ereeere123 opened 7 months ago

ereeere123 commented 7 months ago

I would be extremely grateful.

junjie-shentu commented 4 months ago

Hello! Sorry for the late reply! If you want to use the UViT as backbone, please check the config setting in configs/train_UViT_config.py and run python train_diffusion_clipUViT.py. Moreover, you can generate images by running python sample_diffusion_clipUViT.py. All the parameters you need for image generation is the path to the data file, path for saving the generated images and path to a model check point.