Giter Club home page Giter Club logo

xraygan's Introduction

XRAY-GAN: Conditional and Unconditional Score-guided GAN for Chest X-ray Pathology Classification

Joy Chen, Qingfu Wan

This is the repository for the final project of "Machine Learning - CSCI-GA.2565-001" Spring 2021 at NYU Courant.


Pipelines

CGAN

UCGAN


Features


  • Conditioning the image generation on class labels for a better equilibrium in imbalanced multi-modal data.

  • Metrics-turned-Trainable-Scores + Self-aligned Perceptual Style Scores for better-learned data distribution.

  • Built-in classifier as an implicit evaluator and an explicit learner.

  • Label-aware class-specific generation + class-agnostic homogenous generation system.

  • A foolproof online self-annotation mechanism compensating for insufficient data.


Report


Full report here.


General Structure


${ROOT}
+-- experiments
+-- figs   
+-- code
+-- full_code
+-- README.md

Clean Code

${ROOT}/code/

The Complete Set of Code

${ROOT}/full_code/


Environment


  • PyTorch
  • Keras

GANerated Images


Left: Real Data; Right: Generated Data

CGAN

Clear indication of ribs, spine, and heart shading.

UCGAN

Clear spines, lungs, and varied contrast & color gradient.


Enjoy!~


xraygan's People

Contributors

joyc565 avatar strawberryfg avatar

Stargazers

 avatar  avatar  avatar

Watchers

 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.