FredZhang7 commited on
Commit
bfd24fa
1 Parent(s): 7ef6696

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -4
README.md CHANGED
@@ -12,16 +12,14 @@ metrics:
12
 
13
 
14
  Google Safesearch Mini V2 took a different approach to its training process than V1; it used the InceptionResNetV2 architecture and a dataset of roughly **3,400,000 images** randomly sourced from the internet, some of which were generated via data argumentation.
 
15
  After training the model for 5 epochs with cross entropy loss and evaluating it on both the training and validation sets to identify images with predicted probabilities below 0.90, some necessary corrections were made to the curated dataset and the model was trained for an additional 8 epochs.
16
 
17
  I tested the model on various cases that it may struggle to classify and observed that it was mistaking the fur of a brown cat for human skin.
18
  To improve the accuracy, I fine-tuned the model with 5 extra datasets of animal images from Kaggle for one epoch, and then trained it for the last epoch with a combination of training and test data.
19
  This resulted in **97% accuracy** on both training and validation data.
20
 
21
-
22
- The training and validation data are sourced from Google Images, Reddit, Kaggle, and Imgur, and were classified as safe or nsfw by companies, Google SafeSearch, and moderators.
23
- This model also offers a major benefit over stable diffusion safety checkers - users can save 1.0 GB of RAM and disk space.
24
-
25
 
26
  ## PyTorch
27
  ```bash
 
12
 
13
 
14
  Google Safesearch Mini V2 took a different approach to its training process than V1; it used the InceptionResNetV2 architecture and a dataset of roughly **3,400,000 images** randomly sourced from the internet, some of which were generated via data argumentation.
15
+ The training and validation data are sourced from Google Images, Reddit, Kaggle, and Imgur, and were classified as safe or nsfw by companies, Google SafeSearch, and moderators.
16
  After training the model for 5 epochs with cross entropy loss and evaluating it on both the training and validation sets to identify images with predicted probabilities below 0.90, some necessary corrections were made to the curated dataset and the model was trained for an additional 8 epochs.
17
 
18
  I tested the model on various cases that it may struggle to classify and observed that it was mistaking the fur of a brown cat for human skin.
19
  To improve the accuracy, I fine-tuned the model with 5 extra datasets of animal images from Kaggle for one epoch, and then trained it for the last epoch with a combination of training and test data.
20
  This resulted in **97% accuracy** on both training and validation data.
21
 
22
+ A safesearch filter is not only a great tool for moderating social media, but it also can be used to filter datasets. Compared to stable diffusion safety checkers, this model offers a major advantage - users can save 1.0 GB of RAM and disk space.
 
 
 
23
 
24
  ## PyTorch
25
  ```bash