s3nh / nsfw-classifier

nsfw images classifier using ResNet-18. Pytorch

nsfw-classifier

nsfw images classifier using ResNet-18. Pytorch

Dataset stored in:

-----\sfw

-----\nsfw

Labels provided by pytorch DataLoader.

To do :

results and structure, add section with test dataset. In progress...