STAR: A Structure-aware Lightweight Transformer for Real-time Image Enhancement
STAR-pytorch
Implementation for paper “STAR: A Structure-aware Lightweight Transformer for Real-time Image Enhancement” (ICCV 2021).
STAR-DCE
The pytorch implementation of low light enhancement with STAR on Adobe-MIT FiveK dataset. You can find it in STAR-DCE directory. Here we adopt the pipleline of Zero-DCE ( paper | code ), just replacing the CNN backbone with STAR. In Zero-DCE, for each image the network will regress a group of curves, which will then applied on the source image iteratively. You can find more details in the original repo Zero-DCE.
Requirements
- numpy
- einops
- torch
- torchvision
- opencv
Datesets
We provide download