All Projects → AnjanDutta → Sem Pcyc

AnjanDutta / Sem Pcyc

Licence: mit
PyTorch implementation of the paper "Semantically Tied Paired Cycle Consistency for Zero-Shot Sketch-based Image Retrieval", CVPR 2019.

Programming Languages

python
139335 projects - #7 most used programming language

Projects that are alternatives of or similar to Sem Pcyc

Ali Pytorch
Adversarially Learned Inference in Pytorch
Stars: ✭ 27 (-64.47%)
Mutual labels:  generative-model
Csgnet
CSGNet: Neural Shape parser for Constructive Solid Geometry
Stars: ✭ 55 (-27.63%)
Mutual labels:  generative-model
Torchgan
Research Framework for easy and efficient training of GANs based on Pytorch
Stars: ✭ 1,156 (+1421.05%)
Mutual labels:  generative-model
Dcgan Pytorch
PyTorch Implementation of DCGAN trained on the CelebA dataset.
Stars: ✭ 32 (-57.89%)
Mutual labels:  generative-model
Adain Style
Arbitrary Style Transfer in Real-time with Adaptive Instance Normalization
Stars: ✭ 1,049 (+1280.26%)
Mutual labels:  generative-model
Vae protein function
Protein function prediction using a variational autoencoder
Stars: ✭ 57 (-25%)
Mutual labels:  generative-model
Dcgan Tensorflow
A tensorflow implementation of "Deep Convolutional Generative Adversarial Networks"
Stars: ✭ 6,963 (+9061.84%)
Mutual labels:  generative-model
Dfc Vae
Variational Autoencoder trained by Feature Perceputal Loss
Stars: ✭ 74 (-2.63%)
Mutual labels:  generative-model
Notes
The notes for Math, Machine Learning, Deep Learning and Research papers.
Stars: ✭ 53 (-30.26%)
Mutual labels:  generative-model
Frechet Inception Distance
CPU/GPU/TPU implementation of the Fréchet Inception Distance
Stars: ✭ 66 (-13.16%)
Mutual labels:  generative-model
Pytorch Mnist Vae
Stars: ✭ 32 (-57.89%)
Mutual labels:  generative-model
Ligdream
Novel molecules from a reference shape!
Stars: ✭ 47 (-38.16%)
Mutual labels:  generative-model
Tars
A deep generative model library in Theano and Lasagne
Stars: ✭ 61 (-19.74%)
Mutual labels:  generative-model
Discogan Pytorch
PyTorch implementation of "Learning to Discover Cross-Domain Relations with Generative Adversarial Networks"
Stars: ✭ 961 (+1164.47%)
Mutual labels:  generative-model
Conditional Animegan
Conditional GAN for Anime face generation.
Stars: ✭ 70 (-7.89%)
Mutual labels:  generative-model
Simple Variational Autoencoder
A VAE written entirely in Numpy/Cupy
Stars: ✭ 20 (-73.68%)
Mutual labels:  generative-model
Li emnlp 2017
Deep Recurrent Generative Decoder for Abstractive Text Summarization in DyNet
Stars: ✭ 56 (-26.32%)
Mutual labels:  generative-model
Markov Chain Gan
Code for "Generative Adversarial Training for Markov Chains" (ICLR 2017 Workshop)
Stars: ✭ 76 (+0%)
Mutual labels:  generative-model
Pytorch Pix2pix
Pytorch implementation of pix2pix for various datasets.
Stars: ✭ 74 (-2.63%)
Mutual labels:  generative-model
Dlf
Code for reproducing results in "Generative Model with Dynamic Linear Flow"
Stars: ✭ 65 (-14.47%)
Mutual labels:  generative-model

SEM-PCYC

PyTorch | Arxiv

PyTorch implementation of our SEM-PCYC model for zero-shot sketch-based image retrieval:
Semantically Tied Paired Cycle Consistency for Zero-Shot Sketch-based Image Retrieval
Anjan Dutta, Zeynep Akata
CVPR, 2019

Demo Video

Live Demo

Retrieval Results

Sketchy






TU-Berlin






Prerequisites

  • Linux (tested on Ubuntu 16.04)
  • NVIDIA GPU + CUDA CuDNN
  • 7z
sudo apt-get install p7zip-full

Getting Started

Installation

  • Clone this repository
git clone https://github.com/AnjanDutta/sem-pcyc.git
cd sem-pcyc
  • Install the requirements (not checked)
pip3 install -r requirements.txt
[<host>]
path_dataset = <where all the datasets will be downloaded>
path_aux = <where all the auxiliary folders and files will be saved>

Download datasets

  • Sketchy
  • TU-Berlin
bash download_datasets.sh

Download pretrained models

  • Sketchy
    • sketch
    • image
    • hieremb-jcn + word2vec-google-news
  • TU-Berlin
    • sketch
    • image
    • hieremb-path + word2vec-google-news
bash download_models.sh

Test

Sketchy
python3 src/test.py --dataset Sketchy_extended --dim-out 64 --semantic-models hieremb-jcn word2vec-google-news
TU-Berlin
python3 src/test.py --dataset TU-Berlin --dim-out 64 --semantic-models hieremb-path word2vec-google-news

Train

Sketchy
python3 src/train.py --dataset Sketchy_extended --dim-out 64 --semantic-models word2vec-google-news --epochs 1000 --early-stop 200 --lr 0.0001
TU-Berlin
python3 src/train.py --dataset TU-Berlin --dim-out 64 --semantic-models word2vec-google-news --epochs 1000 --early-stop 200 --lr 0.0001

Citation

@inproceedings{Dutta2019SEMPCYC,
author = {Anjan Dutta and Zeynep Akata},
title = {Semantically Tied Paired Cycle Consistency for Zero-Shot Sketch-based Image Retrieval},
booktitle = {CVPR},
year = {2019}
}

Author

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].