nsfw-detector

Computer Vision project to detect NSFW Images built with Tensorflow in Python

Install libraries

• run pip install -r requirements.txt

Training dataset

• dataset url list in raw data folder obtained from here
• sample scraper script in this notebook

Models preview

Base custom model

Custom model + dropout layer + data augmentation

Transfer learning

Transfer learning + fine tuning