Image Classifiers#
NeMo Curator provides classifiers for image curation, including aesthetic and NSFW classifiers. These models help you filter, score, and curate large image datasets for downstream tasks such as generative model training and dataset quality control.
How It Works#
Image classification in NeMo Curator typically follows these steps:
Generate image embeddings for your dataset (for example, using
TimmImageEmbedder
)Select and configure an image classifier (for example, Aesthetic or NSFW)
Apply the classifier to score or filter images based on the embeddings
Save or further process the classified dataset
You can use built-in classifiers or implement your own for advanced use cases.
Available Classifiers#
Assess the subjective quality of images using a model trained on human aesthetic preferences. Useful for filtering or ranking images by visual appeal.
Detect not-safe-for-work (NSFW) content in images using a CLIP-based classifier. Helps remove or flag explicit material from your datasets.