pe.api package

Subpackages

• pe.api.image package
  • Subpackages
    • pe.api.image.improved_diffusion_lib package
  • Submodules
  • pe.api.image.improved_diffusion_api module
    • ImprovedDiffusion
    • ImprovedDiffusion270M
    • Sampler
    • sample()
  • pe.api.image.stable_diffusion_api module
    • StableDiffusion

Submodules

pe.api.api module

class pe.api.api.API

Bases: ABC

The abstract class that defines the APIs for the synthetic data generation.

abstract random_api(label_info, num_samples)

The abstract method that generates random synthetic data.

Parameters:

• label_info (dict) – The info of the label

• num_samples (int) – The number of random samples to generate

abstract variation_api(syn_data)

The abstract method that generates variations of the synthetic data.

Parameters:

syn_data (pe.data.data.Data) – The data object of the synthetic data

pe.api.util module

class pe.api.util.ConstantList(value)

Bases: object