← 开源
openai

pixel-cnn

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

Model DevelopmentArchitecturePython
在 GitHub 打开
增长势头
+024 小时新增 Star0.0%
1.96k
Star
433
Fork
+-1
本周
8
贡献者
创建于 2016-07-09 · 更新于 2026-10-11 · 今日第 9194 名
主要开发者
karpathydpkingmaTimSalimans

openai/pixel-cnn 排名第 9,194,有 1,956 个星标。

同类别比较

openai/pixel-cnn 排名第 9,194,此处有 8 个同类条目;philipperemy/tensorflow-1.4-billion-password-analysis 排名第 9,196。

同类别比较
名称排名指标
SummitKwan/transparent_latent_gan9,167
语言:
Python
星标数:
1,975
分叉数:
357
简介:
Use supervised learning to illuminate the latent space of GAN for controlled generation and edit
google/nerfies9,169
语言:
Python
星标数:
1,974
分叉数:
277
简介:
This is the code for Deformable Neural Radiance Fields, a.k.a. Nerfies.
google-deepmind/mathematics_dataset9,172
语言:
Python
星标数:
1,971
分叉数:
275
简介:
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
语言:
Python
星标数:
1,959
分叉数:
156
简介:
A collection of robotics simulation environments for reinforcement learning
philipperemy/tensorflow-1.4-billion-password-analysis9,196
语言:
Python
星标数:
1,956
分叉数:
389
简介:
Deep Learning model to analyze a large corpus of clear text passwords.
Fafa-DL/Awesome-Backbones9,201
语言:
Python
星标数:
1,953
分叉数:
269
简介:
Integrate deep learning models for image classification | Backbone learning/comparison/magic modification project
LTH14/mar9,202
语言:
Python
星标数:
1,953
分叉数:
122
简介:
PyTorch implementation of MAR+DiffLoss https://arxiv.org/abs/2406.11838
guillaumegenthial/sequence_tagging9,204
语言:
Python
星标数:
1,952
分叉数:
696
简介:
Named Entity Recognition (LSTM + CRF) - Tensorflow
关键参数与规格
关键参数与规格
参数值
语言Python
类别Model Development
子类别Architecture
星标数1,956
分叉数433
贡献者8
创建日期2016-07-09
更新日期2026-10-11
1 天星标数0
7 天星标数-1
排名9,194
上次排名9,231
排名变化37
7 天星标排名16,988
上次 7 天星标排名16,983
7 天星标排名变化-5
总星标排名6,467
上次总星标排名6,461
总星标排名变化-6
国家USA
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. 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

常见问题

记录了多少星标和分叉?

openai/pixel-cnn 有 1,956 个星标和 433 个分叉。

仓库排名是多少?

openai/pixel-cnn 排名第 9,194。

记录的语言和类别是什么?

openai/pixel-cnn 有 1,956 个星标,语言为 Python,类别为 Model Development。

哪个仓库排名最接近?

openai/pixel-cnn 与 philipperemy/tensorflow-1.4-billion-password-analysis 比较,后者排名第 9,196,有 1,956 个星标。

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

排名相邻
同类项目