image.embedders.base

Module Contents

Classes

ImageEmbedder

API

class image.embedders.base.ImageEmbedder(

model_name: str,

image_embedding_column: str,

classifiers: collections.abc.Iterable[nemo_curator.image.classifiers.ImageClassifier],

)

Bases: abc.ABC

Initialization

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

abstractmethod load_embedding_model(device: str) → collections.abc.Callable