Pixel Cnn

Module: generative_models.models.autoregressive.pixel_cnn

Source: generative_models/models/autoregressive/pixel_cnn.py

Overview

PixelCNN implementation for autoregressive image generation.

Classes

MaskedConv2D

class MaskedConv2D

PixelCNN

class PixelCNN

Functions

call

def __call__()

call

def __call__()

init

def __init__()

init

def __init__()

generate

def generate()

inpaint

def inpaint()

log_prob

def log_prob()

loss_fn

def loss_fn()

Module Statistics

• Classes: 2
• Functions: 8
• Imports: 6