美文网首页数据科学
鉴黄-深度学习Caffe模型之open_nsfw

鉴黄-深度学习Caffe模型之open_nsfw

作者: 南风无影 | 来源:发表于2017-05-17 09:37 被阅读555次

    open_nsfw是yahoo开发的,基于Caffe深度神经网络训练的成人/淫秽图片分类模型。

    NSFW( Not Suitable for Work )– “不适合上班时观看”。

    项目地址:open_nsfw

    输入要检测的图片,它会返回图片评级(0-1),等级越高,图片越黄越暴力。你可以使用它过滤掉尺度较大的图像,0.8以上的过滤掉。

    • 下载open_nsfw
    $ git clone https://github.com/yahoo/open_nsfw
    
    • Build a caffe docker image (CPU)
      Note:git的地址有点错误,要去掉standalone; 如果是gpu版本,把cpu改成gpu即可。
      下面这句命令比较久,看网络情况,公司网络太慢,我花了>3个小时。
    docker build -t caffe:cpu https://raw.githubusercontent.com/BVLC/caffe/master/docker/cpu/Dockerfile
    
    • 检查caffe 的版本号
    docker run caffe:cpu caffe --version
    caffe version 1.0.0-rc3
    
    • 启动Docker,并映射工作目录(open_nsfw),测试图片test_image.jpg
      (图片的读写权限要设置成可读)
    cd open_nsfw
    docker run --volume=$(pwd):/workspace caffe:cpu \
    python ./classify_nsfw.py \
    --model_def nsfw_model/deploy.prototxt \
    --pretrained_model nsfw_model/resnet_50_1by2_nsfw.caffemodel \
    test_image.jpg
    

    测试1:#

    1.jpg

    NSFW 得分: 0.0429051183164##


    测试2:#

    5.jpg

    NSFW 得分:0.941415905952##


    测试3:#

    4.jpg

    NSFW 得分: 0.669750630856##

    相关文章

      网友评论

        本文标题:鉴黄-深度学习Caffe模型之open_nsfw

        本文链接:https://www.haomeiwen.com/subject/abnkxxtx.html