← Open Source
openai

pixel-cnn

Code for the paper "PixelCNN++: A PixelCNN Implementation with Discretized Logistic Mixture Likelihood and Other Modifications".

Model DevelopmentArchitecturePython
Open on GitHub
Momentum
+0stars in 24 hours0.0%
1.96k
Stars
433
Forks
+-1
This week
8
Contributors
Created 2016-07-09 · Updated 2026-10-11 · #9194 today
Top developers
karpathydpkingmaTimSalimans

openai/pixel-cnn is ranked 9,194 with 1,956 stars.

Same-category comparison

openai/pixel-cnn is ranked 9,194 among these 8 peers; philipperemy/tensorflow-1.4-billion-password-analysis is ranked 9,196.

Same-category comparison
NameRankMetrics
SummitKwan/transparent_latent_gan9,167
Language:
Python
Stars:
1,975
Forks:
357
Description:
Use supervised learning to illuminate the latent space of GAN for controlled generation and edit.
google/nerfies9,169
Language:
Python
Stars:
1,974
Forks:
277
Description:
This is the code for Deformable Neural Radiance Fields, a.k.a. Nerfies.
google-deepmind/mathematics_dataset9,172
Language:
Python
Stars:
1,971
Forks:
275
Description:
This dataset code generates mathematical question and answer pairs, from a range of question types at roughly school-level difficulty.
Farama-Foundation/Gymnasium-Robotics9,187
Language:
Python
Stars:
1,959
Forks:
156
Description:
A collection of robotics simulation environments for reinforcement learning.
philipperemy/tensorflow-1.4-billion-password-analysis9,196
Language:
Python
Stars:
1,956
Forks:
389
Description:
Deep Learning model to analyze a large corpus of clear text passwords.
Fafa-DL/Awesome-Backbones9,201
Language:
Python
Stars:
1,953
Forks:
269
Description:
Integrate deep learning models for image classification | Backbone learning/comparison/magic modification project.
LTH14/mar9,202
Language:
Python
Stars:
1,953
Forks:
122
Description:
PyTorch implementation of MAR+DiffLoss https://arxiv.org/abs/2406.11838.
guillaumegenthial/sequence_tagging9,204
Language:
Python
Stars:
1,952
Forks:
696
Description:
Named Entity Recognition (LSTM + CRF) - Tensorflow.
Key parameters and specifications
Key parameters and specifications
ParameterValue
LanguagePython
CategoryModel Development
SubcategoryArchitecture
Stars1,956
Forks433
Contributors8
Created date2016-07-09
Updated date2026-10-11
Stars in 1 day0
Stars in 7 days-1
Rank9,194
Previous rank9,231
Rank change37
7-day stars rank16,988
Previous 7-day stars rank16,983
7-day stars rank change-5
Total stars rank6,467
Previous total stars rank6,461
Total stars rank change-6
CountryUSA
README summary

**Status:** Archive (code is provided as-is, no updates expected) pixel-cnn++ This is a Python3 / Tensorflow implementation of PixelCNN++, as described in the following paper: **PixelCNN++: A PixelCNN Implementation with Discretized Logistic Mixture Likelihood and Other Modifications**, by Tim Salimans, Andrej Karpathy, Xi Chen, Diederik P. Our work builds on PixelCNNs that were originally proposed in van der Oord et al. PixelCNNs are a class of powerful generative models with tractable likelihood that are also easy to sample from

. The core convolutional neural network computes a probability distribution over a value of one pixel conditioned on the values of pixels to the left and above it. Below are example samples from a model trained on CIFAR-10 that achieves **2. 03 of the PixelCNN in van der Oord et al. ): Samples from the model (**left**) and samples from a model that is conditioned on the CIFAR-10 class labels (**right**): This code supports multi-GPU training of our improved

PixelCNN on CIFAR-10 and Small ImageNet, but is easy to adapt for additional datasets. Training on a machine with 8 Maxwell TITAN X GPUs achieves 3. 0 bits per dimension in about 10 hours and it takes approximately 5 days to converge to 2. Setup To run this code you need the following: a machine with multiple GPUs Python3 Numpy, TensorFlow and imageio packages: Training the model Use the script to train the model. To train the default model

on CIFAR-10 simply use: You might want to at least change the and which point to paths on your system to download the data to (if not available), and where to save the checkpoints. To train on fewer GPUs we recommend using to narrow the visibility of GPUs to only a few and then run the script. Don't forget to modulate the flag accordingly. Have a look at the classes in the folder. You have to write an analogous

data iterator object for your own dataset and the code should work well from there. Pretrained model checkpoint You can download our pretrained (TensorFlow) model that achieves 2. Citation If you find this code useful please cite us in your work: pixel-cnn-rotations

Frequently asked questions

How many stars and forks are recorded?

openai/pixel-cnn has 1,956 stars and 433 forks.

What is the repository rank?

openai/pixel-cnn is ranked 9,194.

What language and category are recorded?

openai/pixel-cnn has 1,956 stars, uses Python and belongs to Model Development.

Which repository is closest in rank?

openai/pixel-cnn is compared with philipperemy/tensorflow-1.4-billion-password-analysis, ranked 9,196 with 1,956 stars.

README

Status: Archive (code is provided as-is, no updates expected)

pixel-cnn++

This is a Python3 / Tensorflow implementation of PixelCNN++, as described in the following paper:

PixelCNN++: A PixelCNN Implementation with Discretized Logistic Mixture Likelihood and Other Modifications, by Tim Salimans, Andrej Karpathy, Xi Chen, Diederik P. Kingma, and Yaroslav Bulatov.

Our work builds on PixelCNNs that were originally proposed in van der Oord et al. in June 2016. PixelCNNs are a class of powerful generative models with tractable likelihood that are also easy to sample from. The core convolutional neural network computes a probability distribution over a value of one pixel conditioned on the values of pixels to the left and above it. Below are example samples from a model trained on CIFAR-10 that achieves 2.92 bits per dimension (compared to 3.03 of the PixelCNN in van der Oord et al.):

Samples from the model (left) and samples from a model that is conditioned on the CIFAR-10 class labels (right):

This code supports multi-GPU training of our improved PixelCNN on CIFAR-10 and Small ImageNet, but is easy to adapt for additional datasets. Training on a machine with 8 Maxwell TITAN X GPUs achieves 3.0 bits per dimension in about 10 hours and it takes approximately 5 days to converge to 2.92.

Setup

To run this code you need the following:

  • a machine with multiple GPUs
  • Python3
  • Numpy, TensorFlow and imageio packages:
pip install numpy tensorflow-gpu imageio

Training the model

Use the train.py script to train the model. To train the default model on CIFAR-10 simply use:

python3 train.py

You might want to at least change the --data_dir and --save_dir which point to paths on your system to download the data to (if not available), and where to save the checkpoints.

I want to train on fewer GPUs. To train on fewer GPUs we recommend using CUDA_VISIBLE_DEVICES to narrow the visibility of GPUs to only a few and then run the script. Don't forget to modulate the flag --nr_gpu accordingly.

I want to train on my own dataset. Have a look at the DataLoader classes in the data/ folder. You have to write an analogous data iterator object for your own dataset and the code should work well from there.

Pretrained model checkpoint

You can download our pretrained (TensorFlow) model that achieves 2.92 bpd on CIFAR-10 here (656MB).

Citation

If you find this code useful please cite us in your work:

@inproceedings{Salimans2017PixeCNN,
  title={PixelCNN++: A PixelCNN Implementation with Discretized Logistic Mixture Likelihood and Other Modifications},
  author={Tim Salimans and Andrej Karpathy and Xi Chen and Diederik P. Kingma},
  booktitle={ICLR},
  year={2017}
}

pixel-cnn-rotations

Nearby in the ranking
Similar items