经历种种波折,终于实现了雅虎的open_NSFW项目,记录一下遇到的问题。
项目地址:(第一个是用Tensorflow,第二个是用Caffe)
https://github.com/mdietrichstein/tensorflow-open_nsfw
https://github.com/yahoo/open_nsfw
一、yahoo的open_NSFW中使用到了Caffe,我尝试了很多次都以失败告终,也许是电脑的问题,也许是我自己的问题(我是小白),这期间借鉴了很多大神的帖子,需要的同学自己查看:
深度学习(七)——python3.6环境下使用open_nsfw鉴别色情图片(基于tensorflow)_全部梭哈迟早暴富的博客-CSDN博客_nsfw
二、Caffe安装失败后,发现有人用Tersorflow实现了这个项目,简直像看到了救星(其实并没有想的那么简单),又开始了安装Tensorflow。这个项目支持python3.6 与 tensorflow 1.12,最好安装这个版本,避免不必要的麻烦。
清华镜像用不了?conda安装tensorflow教程以及多种报错的解决(windows环境)_star_xpgz的博客-CSDN博客_conda清华镜像不可用
TensorFlow安装和下载详细教程-内附多种解决方案_Stone的技术栈的博客-CSDN博客_如何下载tensorflow
Windows 安装 Tensorflow 教程_华墨1024的博客-CSDN博客_windows安装tensorflow
下面这个非常有用,如果出现类似报错,只要找到对应文件的行数,把1加上括号和逗号就可以了(不得不佩服解决这些问题的大神们)
源代码是这样:
改成这样:
以上都是从下面这篇文章学到的。
我的安装之路问题重重,
ModuleNotFoundError: No module named 'skimage'(TensorFlow)_gz153016的博客-CSDN博客
终于,解决了各个问题,把Tensorflow安装成功了,开始测试,然而,问题又来了
找到这个文件
把这里的as_grey改成as_gray就可以了,终于可以测试了,为了简单,我把测试图片跟代码放在同一文件夹下,输入
(tensorflow) D:\github\NSFW\tensorflow-open_nsfw-master\tensorflow-open_nsfw-master>python classify_nsfw.py -m data/open_nsfw-weights.npy test9.jpg
就会出现这样的结果(速度有点慢):
花了两天时间,终于能用了,结果出来的那一下还挺激动,都忘了前面解决了多少个问题了,记录一下这一过程,虽然有的问题解决完了已经忘了,如果能帮到其他人那更好了。