StandardCAS-NSTID commited on
Commit
d8922f1
1 Parent(s): 065e0c7

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +11 -1
README.md CHANGED
@@ -24,4 +24,14 @@ We will not share the dataset with any third party libraries. However since AIGC
24
 
25
  Okay now let's get to the topic:
26
 
27
- How can I download the model?
 
 
 
 
 
 
 
 
 
 
 
24
 
25
  Okay now let's get to the topic:
26
 
27
+ Q: How can I download the model?
28
+
29
+ A: You should be able to directly download it from the 'Files' List, aka the "NSFW_Classifier.h5". You can then place it inside the directory of the Estallie-Inference.py file and start the code.
30
+
31
+ Q: How can I train my own model?
32
+
33
+ A: Go to Estallie-trainer.py, edit the Train and Validation folders and model name, then run the program. Remeber the default batch size is 4 so you might want to have more than 4 images per class. Also there is only two classes avaliable for this basic model.
34
+
35
+ Q: How are you supposed to interpret if an image is not NSFW or SFW such as Unethical...etc?
36
+
37
+ A: Good point. At this point we used seperate models to interpret. We also have yolov5 and Deepdanbooru but it's not fun anymore if they're trained for you. We have a backup program for automatically classifying subclasses, the "1c3a.oy" file, and we are trying to complete it.