Skip to content

An implementation of Contourlet Residual for Prompt Learning Enhanced Infrared Image Super-Resolution(CoRPLE)

Notifications You must be signed in to change notification settings

hey-it-s-me/CoRPLE

Folders and files

NameName
Last commit message
Last commit date

Latest commit

b9d0503 · Dec 9, 2024

History

36 Commits
Sep 26, 2024
Sep 12, 2024
Sep 24, 2024
Dec 9, 2024
Sep 12, 2024
Sep 12, 2024

Repository files navigation

Contourlet Residual for Prompt Learning Enhanced Infrared Image Super-Resolution (CoRPLE)

This repo is the official implementation of, “Contourlet Residual for Prompt Learning Enhanced Infrared Image Super-Resolution”, Xingyuan Li, Jinyuan Liu*, Zhixin Chen, Yang Zou, Long Ma, Xin Fan, Risheng Liu, European Conference on Computer Vision (ECCV), 2024.

[pretrained models] [paper link]

Updates

New Version: CRG Branch

The implementation of the 'Contourlet Refinement Gate Framework' is available on the CRG branch.

🤖 Download

Download our datasets of infrared image super-resolution with detection labels. Original images are provided by TarDAL.

Download our datasets of infrared image super-resolution with segmentation labels. Original images are provided by SegMiF.

Dependencies

  • Python 3.8
  • PyTorch 1.8.0
  • NVIDIA GPU + CUDA
# Clone the github repo and go to the default directory 'CoRPLE'.
git clone https://github.com/hey-it-s-me/CoRPLE.git
conda create -n CoRPLE python=3.8
conda activate CoRPLE
pip install -r requirements.txt
python setup.py develop

Training

  • Run the following scripts. The training configuration is in options/train/.
    python basicsr/train.py -opt options/Train/train_CoRPLE_light_x2.yml
    python basicsr/train.py -opt options/Train/train_CoRPLE_light_x4.yml
  • The training experiment is in experiments/.

Testing

  • Run the following scripts. The testing configuration is in options/test/.
    python basicsr/train.py -opt options/Test/my_test_CoRPLE_light_x2.yml
    python basicsr/train.py -opt options/Test/my_test_CoRPLE_light_x4.yml
  • The output is in results/.

Acknowledgements

This code is built on DAT and Contourlet-CNN .

Citation

If this work has been helpful to you, please feel free to cite our paper!

@inproceedings{li2024contourlet,
  title={Contourlet residual for prompt learning enhanced infrared image super-resolution},
  author={Li, Xingyuan and Liu, Jinyuan and Chen, Zhixin and Zou, Yang and Ma, Long and Fan, Xin and Liu, Risheng},
  booktitle={European Conference on Computer Vision},
  pages={270--288},
  year={2024},
  organization={Springer}
}

About

An implementation of Contourlet Residual for Prompt Learning Enhanced Infrared Image Super-Resolution(CoRPLE)

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published