KO | EN

GitLite

Minimalist Open Source Repository Search Engine

gitlite — search

#developer-tools #typescript #claude-code #codex #dsh #dsh-plugin #ai-agent #ai-agents #deepseek-harness #agent-skills #macos #cli

MAT

★ 39 Open GitHub ↗

Official PyTorch implementation of the Motion-adaptive Transformer for Event-based Image Deblurring (AAAI 2025).

// repository documentation

Was this content helpful?

★ 0 (0 ratings)

# MAT: Motion-adaptive Transformer for Event-based Image Deblurring [AAAI 2025]

## Abstract

> Event cameras, which capture pixel-level brightness changes asynchronously, provide rich motion information that is often missed during traditional frame-based camera exposures, thereby offering fresh perspectives for motion deblurring. Although current approaches incorporate event intensity, they neglect essential spatial motion information. Unlike their CNN architectures, Transformers excel in modeling long-range dependencies but struggle with establishing relevant non-local connections in sparse events and fail to highlight significant interactions in dense images. To address these limitations, we introduce a Motion-Adaptive Transformer network (MAT) that utilizes spatial motion information to forge robust global connections. The core design is an Adaptive Motion Mask Predictor (AMMP) that identifies key motion regions, guiding the Motion-Sparse Attention (MSA) to eliminate irrelevant event tokens and enabling the Motion-Aware Attention (MAA) to focus on relevant ones, thereby enhancing long-range dependency modeling. Additionally, we elaborately design a Cross-Modal Intensity Gating mechanism that efficiently merges intensity data across modalities while minimizing parameter use. The learnable Expansion-Controlled Spatial Gating further optimizes the transmission of event features. Comprehensive testing confirms that our approach sets a new benchmark in image deblurring, surpassing previous methods by up to 0.60dB on the GoPro dataset, 1.04dB on the HS-ERGB dataset, and achieving an average improvement of 0.52dB across two real-world datasets.

## Overview

This repository contains the official implementation of our paper "Motion-adaptive Transformer for Event-based Image Deblurring", accepted at the AAAI 2025.

This is the overview of our network's architecture:

![image-20241230103001494](https://xusy-1300242514.cos.ap-nanjing.myqcloud.com/img/image-20241230103001494.png)

See more details in  [[paper]](https://ojs.aaai.org/index.php/AAAI/article/view/32967).

## TODO

- [ ] Release the pretrained models. 
 - [x] Release the visual results of MAT.

## Installation

Use the following command line for the installation of dependencies required to run MAT.

```
git clone https://github.com/QUEAHREN/DemosaicFormer.git
cd DemosaicFormer
conda create -n pytorch181 python=3.7
conda activate pytorch181
conda install pytorch=1.8 torchvision cudatoolkit=10.2 -c pytorch
pip install matplotlib scikit-learn scikit-image opencv-python yacs joblib natsort h5py tqdm
pip install einops gdown addict future lmdb numpy pyyaml requests scipy tb-nightly yapf lpips
```

## Datasets and Visual Results

You can download the datasets processed by us in [[HuggingFace]](https://huggingface.co/datasets/RuixuanJiang/Low_Level_Datasets/blob/main/HS_ERGB_CDS.zip) and visual results of our MAT on GoPro, HS-ERGB, REBlur, and REVD datasets in [[GOOGLE DRIVE]](https://drive.google.com/file/d/1nvC9IuGoJrDBVJ_Vxu07tcWmC8TVoTSW/view?usp=sharing).

## Citation

If you find our work useful in your research, please consider citing:

```
@InProceedings{MAT_2025_AAAI,
    author={Xu, Senyan and Sun, Zhijing and Zhong, Mingchen and Cao, Chengzhi and Liu, Yidi and Fu, Xueyang and Chen, Yan},
    booktitle={Proceedings of the AAAI Conference on Artificial Intelligence},
    volume={39},
    number={9},
    pages={8942--8950},
    year={2025}
}
```

## Contact

Should you have any question, please contact [syxu@mail.ustc.edu.cn](syxu@mail.ustc.edu.cn)

**Acknowledgment:** This code is based on the [BasicSR](https://github.com/xinntao/BasicSR) toolbox.

MAT

Explore Similar Repositories

Was this content helpful?

Recent Feedback

Download README