NavyZeng / Diff-Unmix

Unmixing Diffusion for Self-Supervised Hyperspectral Image Denoising
10 stars 0 forks source link

Request for I190000_E97_gen.pth #6

Open KaKmsyCJY opened 2 weeks ago

KaKmsyCJY commented 2 weeks ago

Dear Professor Zeng, Your Diff-Unmix code in test_diff-unmix_Gaussian_noise.py texted that:

## load model
fix_diff = 1
if fix_diff:
    gen_path = './guided_diffusion/I190000_E97_gen.pth'
    cks = torch.load(gen_path)
    new_cks = OrderedDict()
    for k, v in cks.items():
        newkey = k[11:] if k.startswith('denoise_fn.') else k
        new_cks[newkey] = v
    model.load_state_dict(new_cks, strict=False)
model.to(device)
# model.train()
model.eval()

I think it request I190000_E97_gen.pth and it didn't included. I would be much appreciated it if you could provide this document.
KaKmsyCJY commented 1 week ago

Can I download diffusion-model-I190000_E97_opt.pth to replace this file from https://github.com/wgcban/ddpm-cd/releases?