在生活中不可避免会出现色情视频,因此视频的鉴定就成为了我们需要解决的问题,本博客在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