image.embedders.timm

Module Contents

Classes

TimmImageEmbedder

API

class image.embedders.timm.TimmImageEmbedder(

model_name: str,

pretrained: bool = False,

batch_size: int = 1,

num_threads_per_worker: int = 4,

image_embedding_column: str = 'image_embedding',

normalize_embeddings: bool = True,

classifiers: collections.abc.Iterable = [],

autocast: bool = True,

use_index_files: bool = False,

)

Bases: nemo_curator.image.embedders.base.ImageEmbedder

Initialization

load_dataset_shard(tar_path: str) → collections.abc.Iterable

load_embedding_model(device: str = 'cuda') → torch.nn.Module