All Projects → TobiasSunderdiek → cartoon-gan

TobiasSunderdiek / cartoon-gan

Licence: MIT license
Implementation of cartoon GAN [Chen et al., CVPR18] with pytorch

Programming Languages

Jupyter Notebook
11667 projects
python
139335 projects - #7 most used programming language
Makefile
30231 projects

Projects that are alternatives of or similar to cartoon-gan

deep-explanation-penalization
Code for using CDEP from the paper "Interpretations are useful: penalizing explanations to align neural networks with prior knowledge" https://arxiv.org/abs/1909.13584
Stars: ✭ 110 (+100%)
Mutual labels:  convolutional-neural-network
minirocket
MINIROCKET: A Very Fast (Almost) Deterministic Transform for Time Series Classification
Stars: ✭ 166 (+201.82%)
Mutual labels:  convolutional-neural-network
Cat-Dog-CNN-Classifier
Convolutional Neural Network to classify images as either cat or dog, along with using attention heatmaps for localization. Written in python with keras.
Stars: ✭ 17 (-69.09%)
Mutual labels:  convolutional-neural-network
Hierarchical-Typing
Code and Data for all experiments from our ACL 2018 paper "Hierarchical Losses and New Resources for Fine-grained Entity Typing and Linking"
Stars: ✭ 44 (-20%)
Mutual labels:  convolutional-neural-network
DeTraC COVId19
Classification of COVID-19 in chest X-ray images using DeTraC deep convolutional neural network
Stars: ✭ 34 (-38.18%)
Mutual labels:  convolutional-neural-network
DSMSCN
[MultiTemp 2019] Official Tensorflow implementation for Change Detection in Multi-temporal VHR Images Based on Deep Siamese Multi-scale Convolutional Neural Networks.
Stars: ✭ 63 (+14.55%)
Mutual labels:  convolutional-neural-network
depth-map-prediction
Pytorch Implementation of Depth Map Prediction from a Single Image using a Multi-Scale Deep Network
Stars: ✭ 78 (+41.82%)
Mutual labels:  convolutional-neural-network
Neural Style Tf
TensorFlow (Python API) implementation of Neural Style
Stars: ✭ 2,943 (+5250.91%)
Mutual labels:  convolutional-neural-network
SimpNet-Tensorflow
A Tensorflow Implementation of the SimpNet Convolutional Neural Network Architecture
Stars: ✭ 16 (-70.91%)
Mutual labels:  convolutional-neural-network
Image-Denoising-with-Deep-CNNs
Use deep Convolutional Neural Networks (CNNs) with PyTorch, including investigating DnCNN and U-net architectures
Stars: ✭ 54 (-1.82%)
Mutual labels:  convolutional-neural-network
Keras-MultiClass-Image-Classification
Multiclass image classification using Convolutional Neural Network
Stars: ✭ 48 (-12.73%)
Mutual labels:  convolutional-neural-network
tensorflow-image-classifier
Easily train an image classifier and then use it to label/tag other images
Stars: ✭ 29 (-47.27%)
Mutual labels:  convolutional-neural-network
Preprocessing-Method-for-STEMI-Detection
Official source code of "Preprocessing Method for Performance Enhancement in CNN-based STEMI Detection from 12-lead ECG"
Stars: ✭ 12 (-78.18%)
Mutual labels:  convolutional-neural-network
clinicadl
Framework for the reproducible processing of neuroimaging data with deep learning methods
Stars: ✭ 114 (+107.27%)
Mutual labels:  convolutional-neural-network
deepcurator
A convolutional neural network trained to recognize good* electronic music
Stars: ✭ 38 (-30.91%)
Mutual labels:  convolutional-neural-network
DeepBind-with-PyTorch
CNN architecture for predicting DNA binding sites for Transcription Factors
Stars: ✭ 36 (-34.55%)
Mutual labels:  convolutional-neural-network
sentiment-analysis-of-tweets-in-russian
Sentiment analysis of tweets in Russian using Convolutional Neural Networks (CNN) with Word2Vec embeddings.
Stars: ✭ 51 (-7.27%)
Mutual labels:  convolutional-neural-network
Face.evolve.pytorch
🔥🔥High-Performance Face Recognition Library on PaddlePaddle & PyTorch🔥🔥
Stars: ✭ 2,719 (+4843.64%)
Mutual labels:  convolutional-neural-network
yapic
Yet Another Pixel Classifier (based on deep learning)
Stars: ✭ 24 (-56.36%)
Mutual labels:  convolutional-neural-network
coursera-ai-for-medicine-specialization
Programming assignments, labs and quizzes from all courses in the Coursera AI for Medicine Specialization offered by deeplearning.ai
Stars: ✭ 80 (+45.45%)
Mutual labels:  convolutional-neural-network

CartoonGAN - my attempt to implement it

Within this repo, I try to implement a cartoon GAN [Chen et al., CVPR18].

I created a github page for detailed documentation, please see https://tobiassunderdiek.github.io/cartoon-gan/ for details.

result.png

Usage

Step 1: Generate datasets

All scripts to create the images are resumeable. It is possible to run make cartoons and make photos in parallel by calling them manually in separate terminals.

Cartoon images

  • download all_data.csv from safebooru dataset [2]
  • point to all_data.csv in PATH_TO_SAFEBOORU_ALL_DATA_CSV of cartoon_image_downloader.py
  • run make install to install necessary libraries
  • run make cartoons to download configurable amount of medium size images

Edge-smoothed version of cartoon images

  • run make cartoons-smooth to create the images

Photos

  • download and unzip coco annotations from [3]
  • configure annotations dir location in PATH_TO_COCO_ANNOTATIONS_ROOT_FOLDER of photo_downloader.py
  • run make photos to download configurable amount of photos of persons

Step 2: Train model

All the steps are described in a jupyter notebook on colab, please see here for details.

Step 3: Test

  • run make install-transform
  • download pre-trained weights, they are available for download as part of the release here..
  • run make transform IMAGE=some_example_image_path

Additional information about how to load the pre-trained weights and transform images can be found in the project documentation here: https://tobiassunderdiek.github.io/cartoon-gan/ .

Credits

Thanks to the authors [Chen et al., CVPR18] of the paper for their great work.

References

[Chen et al., CVPR18] http://openaccess.thecvf.com/content_cvpr_2018/papers/Chen_CartoonGAN_Generative_Adversarial_CVPR_2018_paper.pdf

[2] https://www.kaggle.com/alamson/safebooru/download

[3] http://images.cocodataset.org/annotations/annotations_trainval2017.zip

Note that the project description data, including the texts, logos, images, and/or trademarks, for each open source project belongs to its rightful owner. If you wish to add or remove any projects, please contact us at [email protected].