pixel-cnn-pp  by pclucas14

Pytorch implementation of PixelCNN++ for generative image modeling

created 7 years ago
348 stars

Top 80.9% on sourcepulse

GitHubView on GitHub
Project Summary

This repository provides a PyTorch implementation of OpenAI's PixelCNN++ generative model. It is targeted at researchers and practitioners interested in state-of-the-art image generation models, offering a comparable performance to the official TensorFlow implementation.

How It Works

The implementation follows the architecture of PixelCNN++, a powerful autoregressive model that generates images pixel by pixel. It utilizes convolutional neural networks with masked convolutions to ensure causality, predicting each pixel's value based on previously generated pixels. This approach allows for high-quality image synthesis.

Quick Start & Requirements

  • Install via pip install -r requirements.txt.
  • Requires PyTorch, NumPy, Pillow, and tqdm.
  • Pre-trained models are available for download.

Highlighted Details

  • Achieves 2.95 BPD on the test set, closely matching the official TensorFlow implementation's 2.92 BPD.
  • Code structure is maintained for easy comparison with the original OpenAI implementation.

Maintenance & Community

This repository is no longer actively maintained, though issues can still be filed with the expectation of slow responses.

Licensing & Compatibility

The license is not specified in the README. Compatibility for commercial use or closed-source linking is not detailed.

Limitations & Caveats

The implementation omits data-dependent weight initialization and exponential moving average of past models for evaluation, which are present in the official TensorFlow version. The project is explicitly stated as no longer maintained.

Health Check
Last commit

4 years ago

Responsiveness

1 week

Pull Requests (30d)
0
Issues (30d)
0
Star History
1 stars in the last 90 days

Explore Similar Projects

Starred by Patrick von Platen Patrick von Platen(Core Contributor to Hugging Face Transformers and Diffusers), Travis Fischer Travis Fischer(Founder of Agentic), and
3 more.

consistency_models by openai

0.0%
6k
PyTorch code for consistency models research paper
created 2 years ago
updated 1 year ago
Starred by Aravind Srinivas Aravind Srinivas(Cofounder of Perplexity), Ross Taylor Ross Taylor(Cofounder of General Reasoning; Creator of Papers with Code), and
3 more.

pixel-cnn by openai

0.1%
2k
TensorFlow implementation for PixelCNN++ research paper
created 9 years ago
updated 5 years ago
Starred by Lilian Weng Lilian Weng(Cofounder of Thinking Machines Lab), Patrick Kidger Patrick Kidger(Core Contributor to JAX ecosystem), and
4 more.

glow by openai

0.1%
3k
Generative flow research paper code
created 7 years ago
updated 1 year ago
Starred by Andrej Karpathy Andrej Karpathy(Founder of Eureka Labs; Formerly at Tesla, OpenAI; Author of CS 231n), Jiayi Pan Jiayi Pan(Author of SWE-Gym; AI Researcher at UC Berkeley), and
4 more.

taming-transformers by CompVis

0.1%
6k
Image synthesis research paper using transformers
created 4 years ago
updated 1 year ago
Feedback? Help us improve.