pe.embedding.image package

Submodules

pe.embedding.image.inception module

class pe.embedding.image.inception.Inception(res, device='cuda', batch_size=2000)[source]

Bases: Embedding

Compute the Inception embedding of images.

__init__(res, device='cuda', batch_size=2000)[source]

Constructor.

Parameters:

  • res (int) – The resolution of the images. The images will be resized to (res, res) before computing the embedding

  • device (str, optional) – The device to use for computing the embedding, defaults to “cuda”

  • batch_size (int, optional) – The batch size to use for computing the embedding, defaults to 2000

compute_embedding(data)[source]

Compute the Inception embedding of images.

Parameters:

data (pe.data.data.Data) – The data object containing the images

Returns:

The data object with the computed embedding

Return type:

pe.data.data.Data

pe.embedding.image.inception.to_uint8(x, min, max)[source]