README.md

metadata

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)