Hayeong Yu* · Seungjae Han* · Young-Gyu Yoon
(* equal contribution)
We introduce theoretical design principles for self-supervised image denoising and show how they lead to the U-BSN (U-Net Blind Spot Network). U-BSN unifies multi-scale J-invariance with efficient U-Net design, delivering state-of-the-art denoising at a fraction of the cost.
Clone the repository and create an anaconda environment using
pip install -r requirements.txt
We follow the dataset setup in AP-BSN. Please click this link for a detailed preparation description.
To evaluate our model, run:
python test.py -c UBSN_SIDD_val -g 0 --pretrained 'UBSN_SIDD_pretrained.pth' -s UBSN_SIDD_val
python test.py -c UBSN_SIDD_bench -g 0 --pretrained 'UBSN_SIDD_pretrained.pth' -s UBSN_SIDD_bench
-c refers the configuration name (*.yaml),
-g refers the GPU ID,
--pretrained refers to the name of pretrained model file (in the ckpt/ directory),
-s refers to the session name for saving the results.
Download pretrained model in this link, and place the checkpoint as ckpt/UBSN_SIDD_pretrained.pth.
If you find our code or paper useful, please cite
@inproceedings{yu2025design,
title={Design Principles of Multi-Scale J-invariant Networks for Self-Supervised Image Denoising},
author={Yu, Hayeong and Han, Seungjae and Yoon, Young-Gyu},
booktitle={Proceedings of the IEEE/CVF Winter Conference on Applications of Computer Vision},
pages={1309--1318},
year={2025}
}This project is built upon AP-BSN. We thank all the authors for their great work and repos.