Giter Site home page Giter Site logo

teakfi / transfergan Goto Github PK

View Code? Open in Web Editor NEW
0.0 2.0 0.0 170 KB

This is a GAN-system for picture to picture translation, where pixel values in images correspond particle momentum before entering material and after exiting the material.

License: Other

Python 100.00%

transfergan's Introduction

WORK ON THIS HAS CEASED

Currently we are focused on autoencoder based solution.

transferGAN

This is a GAN-system for picture to picture translation, where pixel values in images correspond particle momentum before entering material and after exiting the material.

The data is sparse with only few pixels activated in an 256x256 grid both in before and after images and this will be reflected in the default options, but the codebase should be flexible enough for applying this for other purposes. Another design goal is a modular structure which allows extending this work with newer designs for networks, loss-functions, visualization, and logging.

HDF5 file support

The code has now HDF5 input file support. The data in HDF5 file is supposed to be in root of the file in datasets named: testdata, traindata, rundata, or valdata. At the moment only training is implemented in the code set, so only traindata is being used. The datasets are in a following format: array(index,width,height)=32bit float from 0 to 1, meaning grayscale. Each array has before and after -(image)values, stacked either next to each other or above each other, the default is before on the left and after on the right so the width is double of the single image.

Code ownership

This work is done for Muon-Solutions Oy. The company owns all rights to the code and has agreed to publish it under GPL-3 license. Parts of the code are from BicycleGAN and pytorch-CycleGAN-and-pix2pix and those are owned by their authors.

Thanks

As the codebase grows it will be easy to recognize familirities with the official pytorch implementation of pix2pix and cycleGAN, which is at minimum an inspiration for this code. After some work I realized that BicycleGAN would have been the best starting point, which has the multimodality implemented. The HDF5 loader is based on this posting in Pytorch forum

transfergan's People

Contributors

teakfi avatar

Watchers

Kostas Georgiou avatar  avatar

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.