功能采用雅虎开源框架 open_nsfw_android
使用:
1)开启tflite文件支持
android {
aaptOptions {
noCompress "tflite"
}
}
2)引入依赖
//可选 快速初始化扫描器,可免去初始化代码
implementation 'com.zwy.nsfw:nsfw_initializer:1.3.9'
//必须 扫描器核心文件
implementation 'com.zwy.nsfw:nsfw:1.3.9'
//必须 tensorflow 支持库
implementation 'org.tensorflow:tensorflow-lite:2.1.0'
implementation 'org.tensorflow:tensorflow-lite-gpu:2.1.0'
3)下载模型
下载地址:https://github.com/devzwy/open_nsfw_android/blob/dev/app/src/main/assets/nsfw.tflite
下载之后 文件命名为nsfw.tflite,放在Assets根目录下。
4)初始化
implementation 'com.zwy.nsfw:nsfw_initializer:lastVersion'
引用该库 免去初始化
5)使用
float nsfwScore = NSFWHelper.INSTANCE.getNSFWScore(BitmapFactory.decodeFile(phoneImages.getFile_source())).getNsfwScore();
nsfwScore>0.3 该图片就是可疑图片,nsfwScore可以设置。
使用该库及模型 apk体积增大12M.