|
|
--- |
|
|
title: Nsfw Classify |
|
|
emoji: 🚀 |
|
|
colorFrom: blue |
|
|
colorTo: gray |
|
|
sdk: gradio |
|
|
sdk_version: 3.16.1 |
|
|
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) |
