UCMNet: Uncertainty-Aware Context Memory Network
for Under-Display Camera Image Restoration (CVPR'26)

Daehyun Kim¹ · Youngmin Kim^1,2 · Yoon Ju Oh¹ · Tae Hyun Kim^1†

¹Hanyang University    ²Agency for Defense Development (ADD)

^† Corresponding author

We propose a lightweight Uncertainty-aware Context-Memory Network (UCMNet), for UDC image restoration. Unlike previous methods that apply uniform restoration, UCMNet performs uncertainty-aware adaptive processing to restore high-frequency details in regions with varying degradations.

Installation

Our implementation follows the experimental settings of previous UDC restoration works (e.g., BNUDC and FSI).
Please ensure that scikit-image==0.19.3 is installed.

pip install -r requirements.txt

Data Preparation

POLED: https://yzhouas.github.io/projects/UDC/udc.html

TOLED: https://yzhouas.github.io/projects/UDC/udc.html

SYNTH: https://drive.google.com/drive/folders/13dZxX_9CI6CeS4zKd2SWGeT-7awhgaJF

Pretrained Weights

POLED: ./checkpoints/POLED.pth

TOLED: ./checkpoints/TOLED.pth

SYNTH: ./checkpoints/SYNTH.pth

Evaluation

python testing_n_saving.py

datasets can be converted by option.py.

Visual comparisons on the POLED dataset.

Visual comparisons on the TOLED dataset.

Training

python training_n_recording.py

Citation

@inproceedings{kim2025UCMNet,
  title={UCMNet: Uncertainty-Aware Context Memory Network for Under-Display Camera Image Restoration},
  author={Kim, Daehyun and Kim, Youngmin and Oh, Yoon Ju and Kim, Tae Hyun},
  booktitle={Computer Vision and Pattern Recognition (CVPR)},
  year={2026}
}

Acknowledgement

We gratefully acknowledge the authors of BNUDC and DARKIR for their outstanding work and publicly released code, which laid the foundation for this project.