Deepdive543443 / StarWarp-Lens-distortion-correction-base-on-StarGAN

Codebase of Dissertation Lens distortion correction

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

StarWarp:Lens-distortion-correction-based-on-StarGAN

This repository is the outcome of final year project: StarWarp. This project is a implementation of paper 'The GAN that Warped: Semantic Attribute Editing with Unpaired Data'. It modifies the StarGAN to translate images to warpfields. Warpfields keeps most of the detail of original images. It also could be scaled to apply different degree of editting, or applied on higher resolution.

The effect of this project is still limited due to lack of datasets, time, and computing resources, but it shows this model has potential to be used in photograph enhancing tasks. Includes the lens distortion correction we tried in this project. image

Overview

image

image

image

Environment

python 3.8.10
pytorch(GPU) 1.9.0
albumentations 1.1.0
numpy 1.18.5
tqdm 4.62.4.dev6+g6c930f5
pillow 8.4.0 opencv 4.0.1
opencv-python 4.5.5.62
opencv-python-headless 4.5.5.62

Pretained weight

https://drive.google.com/drive/folders/1lWg_FDx5sDcZkCtbTbiW95itMgaI3kou?usp=sharing

Reference and resource

[For more information see: About](https://arxiv.org/abs/1811.12784)

About

Codebase of Dissertation Lens distortion correction


Languages

Language:Jupyter Notebook 99.4%Language:Python 0.6%