open nsfw封装成接口

来源:互联网 发布:光环大数据培训吃住 编辑:程序博客网 时间:2024/05/18 21:09

承接上文:http://blog.csdn.net/wangqi880/article/details/62037078 黄图识别-open nsfw

为了使用的方便和提供别人好用,想把open nsfw封装成http形式的接口,别人通过上传图片,就可以得到这图片的nsfw score.
由于本人技术有限,python不太熟悉,自己整理了一套方案如下:
1因为图片需要上传到服务器之后,才能调用python的 nsfw,所以这里需要搭建一个ftp服务器。
2使用python的bottle提供http服务,可以调用如下命令进行反馈

python ./classify_nsfw.py \--model_def nsfw_model/deploy.prototxt \--pretrained_model nsfw_model/resnet_50_1by2_nsfw.caffemodel \./pic/h1.jpg

3还需要一个web服务器,来做图片上传,然后调用python的bottole接口 ,处理返回数据。
这里使用httpclient来调用Python的远程接口。
这里写图片描述
web的架构spring+mybatis+springmvc,虽然没有使用到mybatis哈,但是以前有就直接拿来用了。

文件目录:
/home/wangqi//home/wangqi/open_nsfw-master 这里nsfw目录
/home/wangqi//home/wangqi/open_nsfw-master/pic 保存图片目录
可以自己改哈

python的服务脚本就比较简单了:
我是在/home/wangqi/pythonhttp,vim bottoleServer.py
这里写图片描述

import commandsimport bottlefrom bottle import route, run, template@route('/alisa')def index():    params = bottle.request.params    filename=params.get("filename", None)    print filename    command = "python /home/wangqi/open_nsfw-master/classify_nsfw.py --model_def /home/wangqi/open_nsfw-master/nsfw_model/deploy.prototxt --pretrained_model /home/wangqi/open_nsfw-master/nsfw_model/resnet_50_1by2_nsfw.caffemodel /home/wangqi/open_nsfw-master/pic/"+filename    # output = os.popen(command)    return template('<b>{{text}}</b>', text=commands.getoutput(command))run(host='192.168.247.131', port=8888)

然后python bottoleServer.py启动就好了
这里写图片描述
。这里启动好了之后,就从github下载javaweb工程就可以使用了哈.

这里写图片描述

这里写图片描述

还有就是open nsfw调用的自己的模型,会输出一大推日志信息,我是使用sub直接截取的分数。如果不想要日志的话,只有部署caffeonspark自己训练模型来用了,以后尝试做吧。

0 0