image.classifiers.nsfw#

Module Contents#

Classes#

NSFWModel

Normalization

NsfwClassifier

API#

class image.classifiers.nsfw.NSFWModel#

Bases: torch.nn.Module

Initialization

forward(x: torch.Tensor) → torch.Tensor#
class image.classifiers.nsfw.Normalization(shape: list[int])#

Bases: torch.nn.Module

Initialization

forward(x: torch.Tensor) → torch.Tensor#
class image.classifiers.nsfw.NsfwClassifier(
embedding_column: str = 'image_embedding',
pred_column: str = 'nsfw_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#