Commit
·
3ec245a
1
Parent(s):
ef9242c
initial upload
Browse files
README.md
ADDED
@@ -0,0 +1,51 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
---
|
2 |
+
license: cc-by-nc-sa-4.0
|
3 |
+
language:
|
4 |
+
- en
|
5 |
+
tags:
|
6 |
+
- nsfw
|
7 |
+
- content-moderation
|
8 |
+
- focalnet
|
9 |
+
- image-classification
|
10 |
+
- transformers
|
11 |
+
- safetensors
|
12 |
+
---
|
13 |
+
|
14 |
+
# NSFW Image Classification Model - Fine-Tuned FocalNet
|
15 |
+
|
16 |
+
## Overview
|
17 |
+
|
18 |
+
This repository contains a **fine-tuned NSFW image classification model** based on **FocalNet**, optimized for content moderation tasks. The model classifies images into three categories: **Safe, Questionable, and Unsafe**.
|
19 |
+
|
20 |
+
This model is an improved version of **[`MichalMlodawski/nsfw-image-detection-large`](https://huggingface.co/MichalMlodawski/nsfw-image-detection-large)**, which was originally built on **[`microsoft/focalnet-base`](https://huggingface.co/microsoft/focalnet-base)**. It has been further **fine-tuned with additional image data obtained from nostrcheck.me**, enhancing its accuracy and robustness in identifying inappropriate content.
|
21 |
+
|
22 |
+
## Model Details
|
23 |
+
|
24 |
+
- **Base Model:** `microsoft/focalnet-base`
|
25 |
+
- **Fine-Tuned From:** `MichalMlodawski/nsfw-image-detection-large`
|
26 |
+
- **Architecture:** `FocalNetForImageClassification`
|
27 |
+
- **Image Input Size:** `224x224 pixels`
|
28 |
+
- **Classification Labels:**
|
29 |
+
- **Safe**: Appropriate content
|
30 |
+
- **Questionable**: Requires manual review
|
31 |
+
- **Unsafe**: NSFW or inappropriate content
|
32 |
+
- **Training Framework:** `PyTorch`
|
33 |
+
- **Model Format:** `safetensors`
|
34 |
+
|
35 |
+
## Limitations
|
36 |
+
|
37 |
+
- **False Positives/Negatives**: The model is highly accurate but may still produce incorrect classifications.
|
38 |
+
- **Contextual Understanding**: The model analyzes images in isolation, without considering accompanying text or metadata.
|
39 |
+
- **License Restrictions**: This model is released under **CC BY-NC-SA 4.0**, which requires attribution, prohibits commercial use, and mandates sharing under the same license.
|
40 |
+
|
41 |
+
## Acknowledgments
|
42 |
+
|
43 |
+
This model is a fine-tuned version of **[`MichalMlodawski/nsfw-image-detection-large`](https://huggingface.co/MichalMlodawski/nsfw-image-detection-large)**, originally trained on **[`microsoft/focalnet-base`](https://huggingface.co/microsoft/focalnet-base)**. Additional training data was obtained from **nostrcheck.me** to further improve its performance.
|
44 |
+
|
45 |
+
## References
|
46 |
+
|
47 |
+
- [FocalNet Official Repository](https://github.com/microsoft/FocalNet)
|
48 |
+
- [Transformers Documentation](https://huggingface.co/docs/transformers/index)
|
49 |
+
- [Hugging Face Model Hub](https://huggingface.co/models)
|
50 |
+
|
51 |
+
This model is part of an ongoing effort to improve **content moderation through AI**. Contributions and feedback are welcome.
|