huggingface / transformers

🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.
https://huggingface.co/transformers
Apache License 2.0
128.75k stars 25.54k forks source link

Add Restormer #22372

Open tushdon2 opened 1 year ago

tushdon2 commented 1 year ago

Model description

Restormer: Efficient Transformer for High-Resolution Image Restoration was published in CVPR 2022, which introduced a new Vision Transformer based architecture for Image Restoration tasks like Deraining, Motion Deblurring, Defocus Deblurring and Denoising. It reduced the time complexity of Self Attention in Vision Transformers from O(n2) to O(n) by introducing Multi-Dconv Head Transposed Attention. It also introduced Gated-Dconv Feed-Forward Network.

@manyana72 and I would like to add this model to Huggingface.

cc: @NielsRogge

Open source status

Provide useful links for the implementation

Paper, Code Implementation and pretrained model weights

atharvakavitkar commented 1 year ago

Hi @tushdon2, please let me know if and how can I contribute to this model.