图片鉴黄主要思路是通过nsfw.tflite模型文件生成Interpreter,然后通过Interpreter获取python中定义的入口ByteBuffer的张量(Tensor),然后把要鉴别的文件做归一化处理,输入到ByteBuffer中,通过运行Interpreter获取结果即可,其流程如下图所示:
image
2.2 实现步骤
2.2.1 添加依赖
在build.gradle中添加TensorFlow依赖:
implementation 'org.tensorflow:tensorflow-lite:0.0.0-nightly'
implementation 'org.tensorflow:tensorflow-lite-gpu:0.0.0-nightly'
2.2.2 拷贝训练模型文件
将resources目录下的nsfw.tflite文件拷贝到手机SD存储卡上。
2.2.3 加载训练模型文件
public void init() {
File file = new File(Environment.getExternalStorageDirectory() +
2022-05-09 21:03:28
7.19MB
java