Update demo_vanilla_unfolded.pth: stepsize=1/||A^T A||, lower noise
#5
by jtachella - opened
Retrained the vanilla unfolded DRS super-resolution demo with the DRS stepsize initialized as 1/||A^T A||_2 (the theoretically-motivated non-divergent stepsize for the L2 data-fidelity gradient) instead of a fixed 1.0, and with noise_level_img reduced from 0.03 to 0.0075. Test PSNR improved from 20.4 dB (old stepsize=1.0 init, 15 epochs, noise=0.03) to 25.1 dB (new init, 30 epochs, noise=0.0075) on BSDS500 2x super-resolution. Companion script change in examples/unfolded/demo_vanilla_unfolded.py.
jtachella changed pull request status to merged