quentintaranpino commited on
Commit
3ec245a
·
1 Parent(s): ef9242c

initial upload

Browse files
Files changed (1) hide show
  1. README.md +51 -0
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.