Skip to content

devzwy/TensorflowLite-NSFW-Android

Repository files navigation

TensorflowLite-NSFW-Android

一步一步安装TensorFlow、生成.tflite文件、移植android/iOS平台

安装TensorFlow

  1. 安装Anaconda(安装3.7版本)

  2. 安装成功后打开终端输入如下命令创建一个新的虚拟环境,方法是选择 Python 解析器并创建一个 ./venv 目录来存放它:

    conda create -n venv pip python=3.6  # select python version

3.激活虚拟环境:

source activate venv

4.安装TensorFlow

pip install --ignore-installed --upgrade tensorflow==1.13.1

5.克隆open_nsfw项目

6.下载PyCharm编译器

7.导入open_nsfw的项目,这里需要注意,配置idea中python的编译环境为conda中的环境,否则py文件将检测不到tensorflow等相关类

8.修改检测图片绝对路径,运行项目可以得到类似如下值(sfw:适宜在公共场所浏览,nsfw:不适宜在公共场所浏览,两者都是0-1中间的浮点型值,nsfw数值越大表示色情程度越高,sfw反之):
nsfw_img

8.复制nsfw.tflite文件到android项目

9.参考本demo

10.运行demo,获取类似如下结果值(sfw:适宜在公共场所浏览,nsfw:不适宜在公共场所浏览,两者都是0-1中间的浮点型值,nsfw数值越大表示色情程度越高,sfw反之):

11.移动端推荐使用tflite文件,PC端推荐使用pb模型

nsfw_img

About

一步一步安装TensorFlow、生成.tflite文件、移植android/iOS平台

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published