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.