Skip to content

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

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

16 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

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

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published