centos下搭建open_nsfw

服务器yum源为阿里源

open nsfw是雅虎开源的一款开源的黄图识别的模型。
nsfw翻译为不可在工作中看的图片。主要是针对黄图的,恐怖,血腥图片不能识别。
在centos7.2搭建open nsfw的环境

1.安装依赖

sudo yum -y install protobuf-devel leveldb-devel snappy-devel opencv-devel boost-devel hdf5-devel

sudo yum -y install gflags-devel glog-devel lmdb-devel

sudo yum -y install liblas-devel atlas-devel

sudo yum -y install openblas-devel

sudo yum -y install python-devel python-pip

2.安装caffe
cd /home
wget https://github.com/BVLC/caffe/archive/1.0.tar.gz

tar zxvf 1.0.tar.gz

cd /home/caffe-1.0/python/

for req in $(cat requirements.txt);do pip install req; done

cd /home/caffe-1.0

cp Makefile.config.example Makefile.config

vi Makefile.config

因为是虚拟机,这里cpu 模式,所以需要吧CPU_ONLY:=1前面的注释去掉

CPU_ONLY := 1

对于CentOS7,没有libatlas-base-dev这个包,那么可以通过修改编译参数使用openblas解决

修改前
BLAS := atlas
修改后
BLAS := open

修改配置文件完毕后开始编译
make pycaffe
make all

报错:/usr/lib64/libpangoft2-1.0.so.0: undefined reference to `g_log_structured_standard’

解决方法:sudo yum -y update glib2

make test

make runtest

export PYTHONPATH=/home/caffe-1.0/python:$PYTHONPATH

python

报错:ImportError: No module named numpy

解决方法:sudo yum -y install numpy

报错:ImportError: No module named skimage.io

解决方法:sudo yum -y install python2-scikit-image

报错:ImportError: No module named matplotlib

解决方法:sudo yum -y install python-matplotlib

报错:ImportError: No module named google.protobuf.internal

解决方法:pip install protobuf

3安装open nsfw

cd /home/

sudo yum -y install git

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

mkdir image

放几张图片到image目录下

python ./classify_nsfw.py --model_def nsfw_model/deploy.prototxt --pretrained_model nsfw_model/resnet_50_1by2_nsfw.caffemodel ./image/11.jpg

三张图片的数值分别是:

NSFW score: 0.943383216858

NSFW score: 0.765072345734

NSFW score: 0.446755349636

socre < 0.2 很安全
score >= 0.2 && score <= 0.8 介于色情与安全之间
score > 0.8 很大可能性是色情

github上的一些开源数据集:
https://github.com/EBazarov/nsfw_data_source_urls
https://github.com/alexkimxyz/nsfw_data_scraper
保重身体。

猜你喜欢

转载自blog.csdn.net/goodsirlee/article/details/89249234