Augmentation

Module: generative_models.extensions.vision.augmentation

Source: generative_models/extensions/vision/augmentation.py

Overview

Advanced image augmentation for generative models.

This module provides JAX-compatible image augmentation techniques for training robust generative models.

Classes

AdvancedImageAugmentation

class AdvancedImageAugmentation

Functions

call

def __call__()

init

def __init__()

apply_cutout

def apply_cutout()

apply_horizontal_flip

def apply_horizontal_flip()

create_augmentation_sequence

def create_augmentation_sequence()

Module Statistics

• Classes: 1
• Functions: 5
• Imports: 5