ptran1203 / fxgan Public

Notifications You must be signed in to change notification settings
Fork 2
Star 6

Data augmentation for Chestx-ray classification using GAN

6 stars 2 forks Branches Tags Activity

Notifications

Name		Name	Last commit message	Last commit date
Latest commit History 1,221 Commits
data_augmentation		data_augmentation
images		images
notebooks		notebooks
.DS_Store		.DS_Store
.gitignore		.gitignore
bagan.py		bagan.py
batch_gen.py		batch_gen.py
const.py		const.py
dagan.py		dagan.py
fxgan.py		fxgan.py
logger.py		logger.py
metrics.py		metrics.py
readme.md		readme.md
triplet_loss.py		triplet_loss.py
utils.py		utils.py

Repository files navigation

The FX-GAN for chest-xray classification

This repository is implemented based on https://github.com/IBM/BAGAN
To run the experiment please check the file notebooks/fxgan.ipynb

Generator architecture

Discriminator architecture

Experiment results

Data distribution of generated images (feature extracted from pre-trained metric model)

Generated images for 05 seen classes

Generated images for 03 unseen classes

About

Data augmentation for Chestx-ray classification using GAN

gan conditional-gan chest-xrays few-shot-learning

Report repository

Releases

No releases published

Packages

No packages published

Languages