Downloads · 30 days
6
10% of all-time downloads
birder-project/nsfw-predictor
nsfw-predictor is a machine learning model from birder-project. Use it for the machine learning task on the model card, and read the license before you ship it in a product. It is set up for birder. The card lists the license as mit.
A simple MLP intended to run on CLIP embeddings to classify NSFW images.
Downloads · 30 days
6
10% of all-time downloads
All-time downloads
58
Public
Repo size
866 KB
Likes
0
Public
Click a slice to open those files.
.pt866 KB · 100%
From the Hugging Face model README
A simple MLP intended to run on CLIP embeddings to classify NSFW images.
Trained by LAION-AI and only adapted to suit the Vision Data Curation project.
For more information see: https://github.com/LAION-AI/CLIP-based-NSFW-Detector
Original authorship: Adapted from LAION-AI’s CLIP-based-NSFW-Detector
This classifier operates on CLIP image embeddings rather than raw pixels. To run inference with the Birder framework:
# Download the CLIP backbone
python -m birder.tools download-model vit_l14_pn_quick_gelu_openai-clip
# Run prediction on a dataset
python -m birder.scripts.predict \
-n vit_l14_pn_quick_gelu \
-t openai-clip \
--simple-crop \
--gpu \
--parallel \
--batch-size 256 \
--chunk-size 50000 \
--amp \
--amp-dtype bfloat16 \
--save-logits \
--suffix optional-dataset-name \
path/to/dataset
# Can now run the NSFW classifier on the saved logits
Primary use case: Filtering and scoring image embeddings for potentially NSFW content.
Recommended scope: Pre-screening in research, data curation, and large-scale dataset processing.
Not intended for: Deployment as a sole moderation tool, enforcement decisions, or safety-critical applications.
@misc{LAION-AI2022CLIP-based-NSFW-Detector,
author = {Christoph Schuhmann},
title = {CLIP-based-NSFW-Detector},
year = {2022},
url = {https://github.com/LAION-AI/CLIP-based-NSFW-Detector},
note = {Accessed: August 22, 2025},
}