FredZhang7
commited on
Commit
·
1e5dd4d
1
Parent(s):
dae8b5c
Update README.md
Browse files
README.md
CHANGED
@@ -7,7 +7,7 @@ tags:
|
|
7 |
# Google Safesearch Mini Model Card
|
8 |
This model is trained on 2,224,000 images scraped from Google Images, Reddit, Imgur, and Github.
|
9 |
It predicts the likelihood of an image being nsfw_gore, nsfw_suggestive, and safe.
|
10 |
-
After 20 epochs on PyTorch
|
11 |
|
12 |
Using this instead of the stable diffusion safety checker allows users to save 1.12GB of RAM and disk space.
|
13 |
|
|
|
7 |
# Google Safesearch Mini Model Card
|
8 |
This model is trained on 2,224,000 images scraped from Google Images, Reddit, Imgur, and Github.
|
9 |
It predicts the likelihood of an image being nsfw_gore, nsfw_suggestive, and safe.
|
10 |
+
After 20 epochs on PyTorch, the finetuned InceptionV3 model achieves 94% acc on both train and test data. After 3.3 epochs on Keras, the finetuned Xception model scores 94% acc on train set and 92% on test set.
|
11 |
|
12 |
Using this instead of the stable diffusion safety checker allows users to save 1.12GB of RAM and disk space.
|
13 |
|