MMLForge Search models, datasets, tasks… Start free

Home › Models › Image Classification › LukeJacob2023/nsfw-image-detector

N

LukeJacob2023/nsfw-image-detector

Image Classification·LukeJacob2023· 43.2K· 24

transformers apache-2.0 Image Classification 85.8M params dataset:imagefolderbase_model:google/vit-base-patch16-224-in21kbase_model:finetune:google/vit-base-patch16-224-in21klicense:apache-2.0

This model is a fine-tuned version of google/vit-base-patch16-224-in21k on the imagefolder dataset. It achieves the following results on the evaluation set: - Loss: 0.8138 - Accuracy: 0.9316 - Accuracy K: 0.9887

Open in MLForge Sign up free Desktop app Source ↗

# pull & run locally
pip install mlforge-sdk && mlforge pull LukeJacob2023/nsfw-image-detector

Model details

Task

Image Classification

Provider

LukeJacob2023

Framework

transformers

Parameters

85.8M

Size

3.5 GB

License

apache-2.0

Downloads

43.2K

Likes

24

Updated

2024-12-28

About LukeJacob2023/nsfw-image-detector

This model is a fine-tuned version of google/vit-base-patch16-224-in21k on the imagefolder dataset. It achieves the following results on the evaluation set: - Loss: 0.8138 - Accuracy: 0.9316 - Accuracy K: 0.9887

Related Image Classification

M timm/mobilenetv3_small_100.lamb_in1k Image Classification ·2.6M params 22.7M 80 🤗 HF N Falconsai/nsfw_image_detection Image Classification ·85.8M params 9.8M 1.1K 🤗 HF F dima806/fairface_age_image_detection Image Classification ·85.8M params 6.6M 75 🤗 HF V google/vit-base-patch16-224 Image Classification ·86.6M params 5.7M 980 🤗 HF R timm/resnet50.a1_in1k Image Classification 2.6M 43 🤗 HF M apple/mobilevit-small Image Classification 2.4M 91 🤗 HF

Browse all Image Classification models →