Home

Awesome

HMA

Updates

Overview

<img src="https://github.com/korouuuuu/HMA/blob/main/figs/Comparison.png" width="600"/>

Benchmark results on SRx4.

ModelSet5Set14BSD100Urban100Manga109
SwinIR32.9229.0927.9227.4532.03
HMA33.3829.5128.1328.6933.19

Comparison with the state-of-the-art methods. <img src="https://github.com/korouuuuu/HMA/blob/main/figs/Visual_Results.png" width="800"/>

Environment

Install Pytorch first. Then,

pip install -r requirements.txt
python setup.py develop

How To Test

python hma/test.py -opt options/test/HMA_SRx2.yml

The testing results will be saved in the ./results folder.

How To Train

CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7 python -m torch.distributed.launch --nproc_per_node=8 --use_env --master_port=4321 hma/train.py -opt options/train/train_HMA_SRx4_from_Imagenet.yml --launcher pytorch

The training logs and weights will be saved in the ./experiments folder.

Results

The inference results on benchmark datasets are available at Google Drive.

Citations

BibTeX

@InProceedings{Chu_2024_CVPR,
author    = {Chu, Shu-Chuan and Dou, Zhi-Chao and Pan, Jeng-Shyang and Weng, Shaowei and Li, Junbao},
title     = {HMANet: Hybrid Multi-Axis Aggregation Network for Image Super-Resolution},
booktitle = {Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR) Workshops},
month     = {June},
year      = {2024},
pages     = {6257-6266}
}

Contact

If you have any question, please email douzhichao2021@163.com to discuss with the authors.