wgan, wgan2(improved, gp), infogan, and dcgan implementation in lasagne, keras, pytorch

Last update: Dec 16, 2022

Overview

Generative Adversarial Notebooks

Collection of my Generative Adversarial Network implementations

Most codes are for python3, most notebooks works on

CycleGAN

CycleGAN-lasagne
CycleGAN-keras

CycleGAN results

Result after 3 hours and 58 epochs on a GTX 1080. From top to bottom: Input, Fake, Recreate of the input.

Face-off result. From top to bottom: Input, Fake, Recreate of the input. [youtube video](https://www.youtube.com/watch?v=Fea4kZq0oFQ)

pix2pix

pix2pix-keras: pix2pix GAN Keras implementation
pix2pix-lasagne: pix2pix GAN Lasagne implementation
pix2pix-torch: pix2pix GAN pytorch implementation

pix2pix sample results

Validation result of edges-to-shoes after 12 epochs. From top to bottom: Input, Ground truth, the result.

Validation result of facades dataset after 150 epochs using resnet. From top to bottom: Input, Ground truth, the result.

WGAN on CIFAR10

wgan-keras: Wasserstein GAN Keras implementation
wgan-lasagne: Wasserstein GAN Lasagne implementation
wgan-torch: Wasserstein GAN pytorch implementation based on https://github.com/martinarjovsky/WassersteinGAN

WGAN2 (improved WGAN/WGAN-gp)

wgan2-lasagne: improved WGAN Lasagne implementation (on CIFAR10)
wgan2-keras: improved WGAN Keras implementation (on CIFAR10)
wgan2-lasagne-anime: WGAN on anime face images, lasagne
wgan2-AC-lasagne: improved WGAN Lasagne implementation with Auxillary classfier

WGAN2 sample results

cifar10 dataset

cifar10 dataset with Auxillary classfier

anime face dataset

InfoGAN

mnist-infogan: InfoGAN Lasagne on MNIST dataset
mnist-infogan-paper-uniform: InfoGAN Lasagne on MNIST dataset (fllowing the paper implementation)

InfoGAN sample results

DCGAN

dcgan-lasagne: DCGAN in Lasagne

wgan, wgan2(improved, gp), infogan, and dcgan implementation in lasagne, keras, pytorch

Related tags

Overview

Generative Adversarial Notebooks

Collection of my Generative Adversarial Network implementations

CycleGAN

CycleGAN results

pix2pix

pix2pix sample results

WGAN on CIFAR10

WGAN2 (improved WGAN/WGAN-gp)

WGAN2 sample results

InfoGAN

InfoGAN sample results

DCGAN

DCGAN sample results

Owner

tjwei

This repository contains small projects related to Neural Networks and Deep Learning in general.

Annotated, understandable, and visually interpretable PyTorch implementations of: VAE, BIRVAE, NSGAN, MMGAN, WGAN, WGANGP, LSGAN, DRAGAN, BEGAN, RaGAN, InfoGAN, fGAN, FisherGAN

Reference code for the paper CAMS: Color-Aware Multi-Style Transfer.

Implementation for "Seamless Manga Inpainting with Semantics Awareness" (SIGGRAPH 2021 issue)

Official repository for "Intriguing Properties of Vision Transformers" (2021)

SUPERVISED-CONTRASTIVE-LEARNING-FOR-PRE-TRAINED-LANGUAGE-MODEL-FINE-TUNING - The Facebook paper about fine tuning RoBERTa with contrastive loss

Baseline inference Algorithm for the STOIC2021 challenge.

Projecting interval uncertainty through the discrete Fourier transform

ICNet for Real-Time Semantic Segmentation on High-Resolution Images, ECCV2018

Model Quantization Benchmark

A Deep Learning Based Knowledge Extraction Toolkit for Knowledge Base Population

Weakly Supervised Learning of Instance Segmentation with Inter-pixel Relations, CVPR 2019 (Oral)

Episodic-memory - Ego4D Episodic Memory Benchmark

【steal piano】GitHub偷情分析工具！

use machine learning to recognize gesture on raspberrypi

ConformalLayers: A non-linear sequential neural network with associative layers

Implementation of Neonatal Seizure Detection using EEG signals for deploying on edge devices including Raspberry Pi.

Scripts for training an AI to play the endless runner Subway Surfers using a supervised machine learning approach by imitation and a convolutional neural network (CNN) for image classification

The Malware Open-source Threat Intelligence Family dataset contains 3,095 disarmed PE malware samples from 454 families

GrabGpu_py: a scripts for grab gpu when gpu is free