Spaces:
Running
Running
File size: 751 Bytes
7246cbb 3e2ba9b 7246cbb badb516 7246cbb 3e2ba9b |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 |
---
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) |