Skip to content
/ FEGAN Public

Self-Supervised Deep Learning for Fisheye Image Rectification (pytorch implementation).

Notifications You must be signed in to change notification settings

ByronHsu/FEGAN

Repository files navigation

Self-Supervised Deep Learning for Fisheye Image Rectification

This is the official pytorch implementation of Self-Supervised Deep Learning for Fisheye Image Rectification. The code is written by Byron Hsu and Brian Chao from Department of Electrical Engineering, National Taiwan University. This work is currently in the submission to 2020 ICASSP.

We welcome any advice for this work. Feel free to send issue or PR!

Installation

  1. Clone the project
git clone ...
  1. Install the prerequisites
pip3 install -r requirements.txt

Training

  1. Start training!
bash train_gcgan.sh [dataset_path] [name]
  1. Monitor the training process by tensorboard
tensorboard --logdir runs

Testing

bash test_gcgan.sh [dataset_path] [name] [which_epoch]

Reference

Releases

No releases published

Packages

No packages published