The official code for “SimFIR: A Simple Framework for Fisheye Image Rectification with Self-supervised Representation Learning”, ICCV, 2023.
🚀 Demo (Link)
- Upload the distorted document image to be rectified in the left box.
- Click the "Submit" button.
- The rectified image will be displayed in the right box.
This repository contains the dataset organized into two main directories:
- images: Contains distorted image files.
- flow: Stores the warping flow files.
Each of these directories further comprises two sub-directories:
- train: For training data.
- test: For testing data.
Click here to access the dataset.
- Put the pre-trained model to
$ROOT/model_pretrained/. - Put the distorted images in
$ROOT/distorted/. - Run the script and the rectified images are saved in
$ROOT/rectified/by default.python inference.py
If you find this code useful for your research, please use the following BibTeX entry.
@inproceedings{feng2023simfir,
title={Simfir: A simple framework for fisheye image rectification with self-supervised representation learning},
author={Feng, Hao and Wang, Wendi and Deng, Jiajun and Zhou, Wengang and Li, Li and Li, Houqiang},
booktitle={Proceedings of the IEEE/CVF International Conference on Computer Vision},
pages={12418--12427},
year={2023}
}