Stylegan3

Module: generative_models.models.gan.stylegan3

Source: generative_models/models/gan/stylegan3.py

Overview

StyleGAN3-inspired generator and discriminator with simplified style modulation.

This module provides a simplified style-modulated generator/discriminator stack for JAX/Flax NNX with mapping and synthesis networks plus progressive upsampling.

Key Features:

• Style-based generation with mapping and synthesis networks
• Simplified but effective modulated convolutions
• Progressive upsampling architecture
• JAX/Flax NNX compatibility

Classes

MappingNetwork

class MappingNetwork

StyleGAN3Discriminator

class StyleGAN3Discriminator

StyleGAN3Generator

class StyleGAN3Generator

StyleModulatedConv

class StyleModulatedConv

SynthesisBlock

class SynthesisBlock

SynthesisNetwork

class SynthesisNetwork

Functions

call

def __call__()

call

def __call__()

call

def __call__()

call

def __call__()

call

def __call__()

call

def __call__()

init

def __init__()

init

def __init__()

init

def __init__()

init

def __init__()

init

def __init__()

init

def __init__()

sample

def sample()

Module Statistics

• Classes: 6
• Functions: 13
• Imports: 4