Skip to content

Trained tensorflow model for detecting nudity in images

Notifications You must be signed in to change notification settings

minto5050/NSFW-detection

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

4 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Tensorflow NSFW Image classifier

This is the repository developed for 'Image Classifier in TensorFlow in 5 Min on YouTube using this CodeLab by Google as a guide.

Applications can use this model to detect nudity in the images using tensorflow.

Final test accuracy = 98.2% (N=1098)

training command

python retrain.py \ --bottleneck_dir=bottlenecks \ --how_many_training_steps=500 \ --model_dir=inception \ --summaries_dir=training_summaries/basic \ --output_graph=retrained_graph.pb \ --output_labels=retrained_labels.txt \ --image_dir=trainimg

Give a test image to verify the model

python classify.py test_image.jpg

About

Trained tensorflow model for detecting nudity in images

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages