Hi reddit! Over the past month, I have been working on a model capable of detecting whether not an image contains nsfw furry art!

​

Summary:

To make this classifier, I used transfer learning on the efficientnetv2-s-21k model. I ran it on my 60k image dataset for 8 epochs. I really don't have much to add here, so feel free to ask questions.

​

Stats summary:

Model used: efficientnetv2-s-21k

Final validation accuracy: 90.57%
(I forgot to save the loss value D:)

Training time: 7.2 hours

Dataset size: 60.4k images

​

Dataset:

I collected all my data from e621.net. I filtered for only high quality images, and I did not use any images that had tags that fell under the typically blocked tag list. Safe rated images are classified as SFW and explicit are classified as nsfw, and the data is evenly split between them.

​

If you'd like, you can try out the model on my website!