kaaokou/nsfw_pytorch_model

nsfw_pytorch_model

说明

图像分类器，判定图片是否涉黄。

数据主要来源四个地方

• nsfw_data_scraper
• nsfw_data_source_urls-master
• imagenet1000
• Danbooru

	数量	index
drawings	244717	0
hentai	126429	1
neutral	868192	2
porn	294488	3
sexy	5669	4

使用方法

• 见demo.ipynb

模型表现

mobilenet_v3_large_nsfw_base_v1：平均acc 96.05% - 链接:https://pan.baidu.com/s/1eDlG9lVXznEJIdiT8HGdOg 密码:r08g

resnet101_nsfw_base_v1:平均acc 95.09% - 链接:https://pan.baidu.com/s/14srT0wvB4hr0Ba-9-tH4aw 密码:neg8

mobilenet_v3_large_nsfw_balance_v1: # 添加平衡数据处理训练结果 - 链接:https://pan.baidu.com/s/1-V5kaqGp-Oo0ycHJA_LsYA 密码:jwjr