Draft project for nudity detection with simple convolutional neural network in Keras
- Clone repo.
- Choose class mode in congif.yaml and set path to model.
- Run model on any image:
python predict.py --img_path image.jpg
- You can evaluate model on dataset:
python eval.py
- You can train model on dataset:
python train.py
| Class_mode | Accuracy |
|---|---|
| 2 classes (erotic_and_porn, legal) | 0.8946 |
| 3 classes (erotic, porn, legal) | 0.7300 |
- Legal - images without naked people
- Erotic - images with half-naked people (for example, in swimsuits)
- Porn - images with completely naked people or sex scenes