search

Dmsw / RGB-IMAGES-ENHANCING-HYPERSPECTRAL-IMAGE-DENOISING-WITH-DIFFUSION-MODEL

1 stars 0 forks source link

about code running #1

Open ZhangKZ11 opened 1 month ago

ZhangKZ11 commented 1 month ago

Hello, I recently read your article. How should the code be run?

Dmsw commented 1 month ago

the environment is the same as guided-diffusion-model. you can run the script in run/*.sh using bash. However, you may need to modify the --base_samples'' in the arguments, the classHSIDataset'' and function load_hsi_data'' in the fileimage_datasets.py'' to meet your requirements.

ZhangKZ11 commented 1 month ago

the environment is the same as guided-diffusion-model. you can run the script in run/*.sh using bash. However, you may need to modify the --base_samples'' in the arguments, the classHSIDataset'' and function load_hsi_data'' in the fileimage_datasets.py'' to meet your requirements.

Thanks for your prompt reply. How did you train your RGB diffusion and how did you use the "256x256_diffusion_uncond.pt" model mentioned?

Dmsw commented 4 weeks ago

Sorry for delay. the RGB DM is directly downloaded from https://github.com/openai/guided-diffusion.git using the "256x256_diffusion_uncond.pt". the corresponding config is corrected now in the file config\model_config_rgb.yaml.