Giter Club home page Giter Club logo

cycle-gan-for-depth-map-generation's Introduction

Cycle-GAN-for-Depth-Map-Generation

Generative Adversarial Networks have been improved further to overcome the shortcomings that it initially faced. Cycle is one of the variations that combines the properties of conditional constraints and cycle consistencies to effectively use the GAN architecture in Image-to-Image translation tasks. Hence, this was chosen as the technique to generate the depth images for our project. Unlike GANs, cycle-GANs can be used to alter a given input image to a distribution of the target domain. Concretely, an image is taken from input domain D i and then transformed into an image of target domain D t without having a one-to-one mapping between images from the input to target domain in the training set. Since this architecture has this relaxation of one- to-one mapping, it becomes quite powerful as the same method could be employed to tackle variety of problems by varying input and output domain pairs. Since, this method works on unpaired dataset, it becomes more modular than Pix2Pix architecture. This modularity has been achieved by two step transformations- first by mapping the input image to the target domain and then getting back the original image form the target domain, Mapping the image to target domain is done using a generator network and the quality of image is checked by the discriminator which constantly pushes the generator to perform better. alt text

Dependencies

  1. Pytorch 0.4.1 or newer
  2. Python 3.6
  3. Matplotlib 3.0
  4. OpenCV 3.4 or higher

Directory Structure

Dataset directory

alt text

Corresponding images should go in final directory as shown above.

checkpoints_cyclegan directory

Create a directory named checkpoints_cyclegan to save the .pkl files

output directory

Create a directory named output to save the inference outputs

samples_cyclegan directory

Create a directory named samples_cyclegan to save samples while training

Training

cd to the folder where the scripts are placed and run python3 Training.py

Testing

run python3 testing.py

Pre-Trained weights

Pre-trained weights can be found at Pre-Trained Weights

cycle-gan-for-depth-map-generation's People

Contributors

vineet-pandey avatar

Stargazers

 avatar YunYuLai avatar YyangJJuan avatar  avatar  avatar zhanghe avatar  avatar UglyStupidHonest avatar  avatar Niamh  avatar  avatar  avatar

Watchers

James Cloos avatar

Forkers

brucelai

cycle-gan-for-depth-map-generation's Issues

Pretrain Model with .pth format

Hi, Thank you for the great work.

I'm wondering if it is possible that you could share the pretrain model for CycleGAN with .pth format? Due to the limited computational power, I can only train the model on Colab (it only accepts .pth files).

Any help is much appreciated:)

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    ๐Ÿ–– Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. ๐Ÿ“Š๐Ÿ“ˆ๐ŸŽ‰

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google โค๏ธ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.