Edge-Focused Super-Resolution for Omnidirectional Images with Spherical Geometric Augmentation

Pytorch implementation for “Edge-FocusedSuper-ResolutionforOmnidirectionalImageswithSpherical GeometricAugmentation”

Dependencies

Python=3.11.5, PyTorch=2.1.1, numpy, cv2, tqdm

Test scripts

Put dataset in ./dataset.

Put pre-trained model in ./checkpoints.

Run the test.py:

python test.py

ODI-SR and testing datasets

Baidu Cloud:

link: https://pan.baidu.com/s/1X73RzDopBUo2y8PB80T42A?pwd=fbtg

Pretrained model

Baidu Cloud:

link: https://pan.baidu.com/s/1NiLoLJE8DZ4DD9dk4VRgTg?pwd=6jiv

Citation

If you find our paper or code useful for your research, please cite:

Contact

If you have any problem, please contact with me through email. I will reply soon.

My email: wsl537527@email.swu.edu.cn