nsfw-image-detector

<!-- This model card has been generated automatically according to the information the Trainer had access to. You should probably proofread and complete it, then remove this comment. -->

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

You can access 384 version on:

https://huggingface.co/LukeJacob2023/nsfw-image-detector-384

Model description

Labels: ['drawings', 'hentai', 'neutral', 'porn', 'sexy']

Intended uses & limitations

Training and evaluation data

A custom dataset about 28k images, if you need to improve your domain's accurate, you can contribute the dataset to me.

Training procedure

Training hyperparameters

The following hyperparameters were used during training:

• learning_rate: 2e-05
• train_batch_size: 32
• eval_batch_size: 32
• seed: 42
• optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
• lr_scheduler_type: linear
• lr_scheduler_warmup_steps: 500
• num_epochs: 10
• mixed_precision_training: Native AMP

Training results

Framework versions

• Transformers 4.36.2
• Pytorch 2.0.0
• Datasets 2.15.0
• Tokenizers 0.15.0