wwd-ustc/RDTR

Model-aware Representation Learning for Radial Distortion Rectification

Inference

1. 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/.
2. Put the distorted images in $ROOT/distorted/.
3. Distortion rectification. The rectified images are saved in $ROOT/result/ by default.

   python inference.py
   

Acknowledgement

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