Skip to content

kaaokou/nsfw_pytorch_model

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

5 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

nsfw_pytorch_model

说明

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

数据主要来源四个地方

数量 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

About

No description, website, or topics provided.

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

 
 
 

Contributors