image.classifiers.aesthetic#

Module Contents#

Classes#

AestheticClassifier

MLP

API#

class image.classifiers.aesthetic.AestheticClassifier(
embedding_column: str = 'image_embedding',
pred_column: str = 'aesthetic_score',
batch_size: int = -1,
model_path: str | None = None,
)#

Bases: nemo_curator.image.classifiers.base.ImageClassifier

Initialization

load_model(device: str) → torch.nn.Module#
postprocess(series: cudf.Series) → cudf.Series#
class image.classifiers.aesthetic.MLP(input_size: int, xcol: str = 'emb', ycol: str = 'avg_rating')#

Bases: torch.nn.Module

Initialization

forward(x: torch.Tensor) → torch.Tensor#