Skip to content
forked from MegEngine/NBNet

NBNet: Noise Basis Learning for Image Denoising with Subspace Projection

License

Notifications You must be signed in to change notification settings

yifanchengit/NBNet

 
 

Repository files navigation

Credit

Dependencies

  • MegEngine >= 1.3.1 (For DistributedDataParallel)

Training

To re-run the traning, change the directory path in the corresponding .ipynb file.

Original and Modified NBNet Training

Refer the file SYDE675Project.ipynb

Improvement Code of NBNet

Refer the file mod/model.py

Other Techniques

Refer the file KSVD_PCA.ipynb

PSNR Calculation

Refer the file mod/psnr_build.py

Results

Visualized Traing Results

Refer the folder Fig.

Quantized Traing Results (PSNR)

Refer the folder psnr_results.

About

NBNet: Noise Basis Learning for Image Denoising with Subspace Projection

Resources

License

Stars

Watchers

Forks

Packages

No packages published

Languages

  • Jupyter Notebook 93.5%
  • Python 6.5%