Model-aware Representation Learning for Radial Distortion Rectification

Inference

For barrel distortion rectification, download the pretrained models from Baidu Cloud(Extraction code: 76fx). For pincushion distortion rectification, download the pretrained models from Baidu Cloud(Extraction code: fabg). Put the model to $ROOT/model_pretrained/.
Put the distorted images in $ROOT/distorted/.
Distortion rectification. The rectified images are saved in $ROOT/result/ by default.
```
python inference.py
```

The codes are largely based on PCN and Blind. Thanks for their wonderful works.

Name		Name	Last commit message	Last commit date
Latest commit History 17 Commits
distorted		distorted
README.md		README.md
dataset.py		dataset.py
dataset_pretraning.py		dataset_pretraning.py
extractor.py		extractor.py
inference.py		inference.py
model.py		model.py
model_pretraning.py		model_pretraning.py
position_encoding.py		position_encoding.py