README.md

---
title: Nsfw Classify
emoji: 🚀
colorFrom: blue
colorTo: gray
sdk: gradio
sdk_version: 5.15.0
app_file: app.py
pinned: false
---

模型数据集来自https://github.com/alex000kim/nsfw_data_scraper
同时我还自己爬取了将近10g的资料集。预训练resnet18在训练集上跑到95% acc便停止训练，因为本身数据集噪音较大，acc过高可能出现过拟合。
下面是摘选自原数据集对各个标签的定义：
porn - pornography images
hentai - hentai images, but also includes pornographic drawings
sexy - sexually explicit images, but not pornography. Think nude photos, playboy, bikini, etc.
neutral - safe for work neutral images of everyday things and people
drawings - safe for work drawings (including anime)