基于深度学习的色情视频鉴定

在生活中不可避免会出现色情视频，因此视频的鉴定就成为了我们需要解决的问题，本博客在NSFW项目（见下面参考文献）的基础上面改进了封装，用来检测视频是否是色情视频。首先，这个项目是基于Caffe 的，使用的网络结构是ResNet网络（可以查看参考文献中的论文）。

为了完成对视频的检测，博主使用了FFMPEG，用来从视频中提取帧数，每20秒提取一次图像，当然为了检测更加精确，可以在后面修改间隔时间。

检测分为三个等级，score < 0.2 的表示很安全，socre > 0.8 的表示有很大的可能性是色情的。

最后程序输出：

总共提取检测视频中的图像帧数

socre < 0.2   很安全 safe的数量，占的比重 

score >= 0.2 && score <= 0.8  medimum ， 介于危险和安全之间的数量，比重

score > 0.8 dangerous， 有很大可能性是色情占的比重

最后我们可以根据dangerous 占的比重可以确定视频是否是色情视频。

废话不多说，下面进入实战环节。

首先就是安装ffmpeg，由于我使用的是Ubuntu 14的系统，安装这个的时候着实费了一点功夫，所幸终于找到了一个好用的源安装成功。

sudo add-apt-repository ppa:mc3man/trusty-media
sudo apt-get update
sudo apt-get install ffmpeg gstreamer0.10-ffmpeg

接下来就是安装caffe了，如果你没有安装过，也没有关系，使用docker就可以了。

安装docker容器：

这里就不多说了。

安装caffe 容器 CPU版：

docker build -t caffe:cpu https://raw.githubusercontent.com/BVLC/caffe/master/docker/cpu/Dockerfile

查看caffe版本：

docker run caffe:cpu caffe --version

下载open_nsfw：

 git clone https://github.com/yahoo/open_nsfw

进入工作目录：

cd open_nsfw

在这里需要说明的是启动docker的时候需要我们把工作目录挂载到docker中，例如：

docker run -ti --volume={}:/workspace caffe:cpu bash".format("/home/duan/open_nsfw/

这里先不用着急运行这一步。

接下来就是视频帧提取代码了，每20秒提取一帧，存放于open_nsfw/picture文件夹下面：

# -*- encoding:utf-8 -*-
__date__ = "17/1/16"
__author__  = "duan"


import os
import shutil
from argparse import Argume