Home

Awesome

ExpoMamba

<p align="center"> <img src="assets/expomamba-logo.png" width="200"> </p>

This is the official repository of ExpoMamba: Exploiting Frequency SSM Blocks for Efficient and Effective Image Enhancement. (ICML ES-FoMo 2024) Eashan Adhikarla, Kai Zhang, John Nicholson, Brian D. Davison.

[x] Code will be released on Sept 1, 2024. [x] Checkpoint release. [] Out of Distribution Lowlight training checkpoint.

Checkpoint

SICE Checkpoint: Dropbox link

BibTeX Reference

If you found our paper and code useful, please cite the following paper:

@inproceedings{
  adhikarla2024expomamba,
  title={ExpoMamba: Exploiting Frequency {SSM} Blocks for Efficient and Effective Image Enhancement},
  author={Eashan Adhikarla and Kai Zhang and John Nicholson and Brian D. Davison},
  booktitle={Workshop on Efficient Systems for Foundation Models II @ ICML2024},
  year={2024},
  url={https://openreview.net/forum?id=X9L6PatYhH}
}

Other related works

@article{adhikarla2024unified,
  title={Unified-EGformer: Exposure Guided Lightweight Transformer for Mixed-Exposure Image Enhancement},
  author={Adhikarla, Eashan and Zhang, Kai and VidalMata, Rosaura G and Aithal, Manjushree and Madhusudhana, Nikhil Ambha and Nicholson, John and Sun, Lichao and Davison, Brian D},
  journal={arXiv preprint arXiv:2407.13170},
  year={2024}
}