wyhuai / DDNM

[ICLR 2023 Oral] Zero-Shot Image Restoration Using Denoising Diffusion Null-Space Model
MIT License
1.16k stars 83 forks source link

对单一图片进行去噪 #80

Open AuroraH1 opened 1 week ago

AuroraH1 commented 1 week ago

请问怎样对指定图像进行去噪并保存?好像不得不对imagenet_var_1k文件进行修改。以下是我的代码: python main.py --ni --config imagenet_256.yml --path_y ./Pics/flower123.jpg --eta 0.85 --deg "denoising" --deg_scale 1 --sigma_y 0.2 -i Pics --add_noise

INFO - main.py - 2024-11-15 09:55:15,821 - Using device: cuda Run SVD-based DDNM. 100 sampling steps. travel_length = 1, travel_repeat = 1. Task: denoising. building dataset from exp\imagenet_val_1k.txt read meta done Dataset has size 8 Start from 0 100%|███████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 100/100 [00:19<00:00, 5.24it/s] 100%|███████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 100/100 [00:17<00:00, 5.73it/s] 100%|███████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 100/100 [00:17<00:00, 5.72it/s] 100%|███████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 100/100 [00:17<00:00, 5.66it/s] 100%|███████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 100/100 [00:17<00:00, 5.71it/s] 100%|███████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 100/100 [00:17<00:00, 5.71it/s] 100%|███████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 100/100 [00:17<00:00, 5.71it/s] 100%|███████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 100/100 [00:17<00:00, 5.70it/s] PSNR: 31.00: 100%|██████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 8/8 [02:22<00:00, 17.80s/it] Total Average PSNR: 31.00 Number of samples: 8 并且不知道图片保存到哪里去了