该项目从yahoo-open_nsfw移植，本人能力有限，停止维护该库，部分图片识别可能不准确，仅能保证pb模型识别的结果与tflite结果一致，误差不超过0.00001
所有相关数据已放在data目录下，需要自取
⚠️ ⚠️ ⚠️ 已从jCenter仓库迁移到Maven,原引用地址发生变更，建议所有老用户升级到新版本。新版本需要手动下载nsfw.tflite模型初始化使用
iOS请参考：issues13
Java参考:Tensorflow-Api
C++参考:TensorflowLite-Api
JavaScript参考JS相关文档
其中Python、C++均有两种数据喂入的方式，可根据需求选择pb模型或tfLite文件，详细请参考上面的链接.Java的目前只能加载pb模型。其他的平台可自行百度
- 🔥 下载模型文件并放入assets目录下
- 🔥 开启tflite文件读取支持(解决模型放在assets目录下无法读取的问题。如果模型不放在assets目录下可跳过该步骤)
android {
...
aaptOptions {
//防止模型被压缩
noCompress "tflite"
}
}
- 🔥 引入依赖
dependencies {
...
implementation 'io.github.devzwy:nsfw:1.5.1'
}
- 🔥 初始化（建议在Application中进行）
class KtApp : Application() {
override fun onCreate() {
super.onCreate()
//开启日志输出，可选
NSFWHelper.openDebugLog()
//扫描前必须初始化
NSFWHelper.initHelper(
context = this)
//初始化api原型
/* NSFW初始化函数 内部日志默认关闭，调试环境可使用openDebugLog()开启日志*/
fun initHelper(
context: Context, //建议传入application,传入activity可能会有内存泄漏
modelPath: String? = null,//模型文件路径，为空时将默认从Assets下读取
isOpenGPU: Boolean = true, //是否开启GPU扫描加速，部分机型兼容不友好的可关闭。默认开启
numThreads: Int = 4 //扫描数据时内部分配的线程 默认4
)
}
}
- 🔥 支持的api列表 带返回值的为同步，传入函数的为异步：
异步识别文件： getNSFWScore(file: File, onResult: ((NSFWScoreBean) -> Unit))
异步识别文件： getNSFWScore(filePath: String, onResult: ((NSFWScoreBean) -> Unit))
同步识别bitmap： getNSFWScore(bitmap: Bitmap): NSFWScoreBean
异步识别bitmap： getNSFWScore(bitmap: Bitmap, onResult: ((NSFWScoreBean) -> Unit))
- 🔥 识别结果说明
NSFWScoreBean.sfw ... 非涉黄数值 数值越大约安全
NSFWScoreBean.nsfw ... 涉黄数值 数值越大约危险
NSFWScoreBean.timeConsumingToLoadData ... 装载数据耗时 单位ms
NSFWScoreBean.timeConsumingToScanData ... 扫描图片耗时 单位ms
- 🔥 调用参考
//异步方式
NSFWHelper.getNSFWScore(item.bitmap) {
this.text =
"nsfw:${it.nsfwScore}\nsfw:${it.sfwScore}\n扫描耗时：${it.timeConsumingToScanData} ms"
if (it.nsfwScore > 0.7) {
this.setBackgroundColor(Color.parseColor("#C1FF0000"))
} else if (it.nsfwScore > 0.5) {
this.setBackgroundColor(Color.parseColor("#C1FF9800"))
} else {
this.setBackgroundColor(Color.parseColor("#803700B3"))
}
}
//同步方式
NSFWHelper.getNSFWScore(item.bitmap).let {
this.text =
"nsfw:${it.nsfwScore}\nsfw:${it.sfwScore}\n扫描耗时：${it.timeConsumingToScanData} ms"
if (it.nsfwScore > 0.7) {
this.setBackgroundColor(Color.parseColor("#C1FF0000"))
} else if (it.nsfwScore > 0.5) {
this.setBackgroundColor(Color.parseColor("#C1FF9800"))
} else {
this.setBackgroundColor(Color.parseColor("#803700B3"))
}
}
🔥 安卓手机直接点我安装
