*** layout: overview slug: nemo-curator/nemo\_curator/models/aesthetics title: nemo\_curator.models.aesthetics -------------------------------------- ## Module Contents ### Classes | Name | Description | | -------------------------------------------------------------------- | ----------------------------------------------------------- | | [`AestheticScorer`](#nemo_curator-models-aesthetics-AestheticScorer) | Public interface for aesthetic scoring of video embeddings. | | [`MLP`](#nemo_curator-models-aesthetics-MLP) | Multi-layer perceptron. | ### Data [`_AESTHETICS_MODEL_ID`](#nemo_curator-models-aesthetics-_AESTHETICS_MODEL_ID) [`_AESTHETICS_MODEL_REVISION`](#nemo_curator-models-aesthetics-_AESTHETICS_MODEL_REVISION) ### API ```python class nemo_curator.models.aesthetics.AestheticScorer( model_dir: str ) ``` **Bases:** [ModelInterface](/nemo-curator/nemo_curator/models/base#nemo_curator-models-base-ModelInterface) Public interface for aesthetic scoring of video embeddings. This class provides a standardized interface for scoring the aesthetic quality of video embeddings using a pre-trained model. Get the model ID names associated with this aesthetic scorer. ```python nemo_curator.models.aesthetics.AestheticScorer.__call__( embeddings: torch.Tensor | numpy.typing.NDArray[numpy.float32] ) -> torch.Tensor ``` Score the aesthetics of input embeddings. **Parameters:** Input embeddings as either a torch tensor or numpy array. **Returns:** `torch.Tensor` Aesthetic scores for each input embedding. ```python nemo_curator.models.aesthetics.AestheticScorer.download_weights_on_node( model_dir: str ) -> None ``` classmethod Download the weights for the aesthetic scorer on the node. ```python nemo_curator.models.aesthetics.AestheticScorer.get_weights_path() -> str ``` Get the path to the weights for the aesthetic scorer. ```python nemo_curator.models.aesthetics.AestheticScorer.setup() -> None ``` Set up the aesthetic scoring model by loading weights. ```python class nemo_curator.models.aesthetics.MLP() ``` **Bases:** `Module` Multi-layer perceptron. A neural network that processes embeddings to predict aesthetic scores. ```python nemo_curator.models.aesthetics.MLP.forward( embed: torch.Tensor ) -> torch.Tensor ``` Forward pass through the MLP. **Parameters:** Input embeddings tensor. **Returns:** `torch.Tensor` Predicted aesthetic scores. ```python nemo_curator.models.aesthetics._AESTHETICS_MODEL_ID = 'ttj/sac-logos-ava1-l14-linearMSE' ``` ```python nemo_curator.models.aesthetics._AESTHETICS_MODEL_REVISION = '1e77fa0' ```