lakshaychhabra / NSFW-Detection-DL Public

Notifications You must be signed in to change notification settings
Fork 10
Star 44

This repo contains Deep learning Implementation for identifying NSFW images.

44 stars 10 forks Branches Tags Activity

Notifications

Name		Name	Last commit message	Last commit date
Latest commit History 9 Commits
.ipynb_checkpoints		.ipynb_checkpoints
.gitignore		.gitignore
Final_weights.h5		Final_weights.h5
Frame by Frame NSFW.ipynb		Frame by Frame NSFW.ipynb
Frame by Frame NSFW.pdf		Frame by Frame NSFW.pdf
NSFW-VIDEO- DETECTOR.ipynb		NSFW-VIDEO- DETECTOR.ipynb
NSFW-VIDEO- DETECTOR.pdf		NSFW-VIDEO- DETECTOR.pdf
NSFW.ipynb		NSFW.ipynb
NSFW.pdf		NSFW.pdf
README.md		README.md
final.png		final.png

Repository files navigation

NSFW Classifier

The NSFW Classifier is used to Classify Not safe for work images vs Safe images. NSFW images includes Porn and Sexy images. Rest are classified as Safe Images

Take a look at a Example

Take a look at a Example which predicts various images in a Single big image

You can see in Frame by Frame file we can implement a function which tells if video is safe for kids or not and We can take actions accordingly.

Mobile Net Architecture is used for the classification as it is very fast and has less params to train.

This is inspired by https://www.freecodecamp.org/news/how-to-set-up-nsfw-content-detection-with-machine-learning-229a9725829c/ , https://github.com/GantMan/nsfw_model The Data for this project is collected with help of scripts at https://github.com/alex000kim/nsfw_data_scraper