色情图片离线识别,基于TensorFlow实现。识别只需200ms,可断网测试,成功率99%,调用只要一行代码,从雅虎的开源项目open_nsfw_python移植,tflite(6M)为训练好的模型,该模型文件可用于iOS、java、C++等平台
- Add it in your root build.gradle at the end of repositories:
allprojects {
repositories {
...
maven { url 'https://jitpack.io' }
}
}
dependencies {
//versionCode:上面小icon中最新版本号
implementation 'com.github.devzwy:open_nsfw_android:[versionCode]'
}
- 扫描时报如下错误
java.lang.NullPointerException: Attempt to invoke virtual method 'com.zwy.nsfw.api.NsfwBean com.zwy.nsfw.Classifier.run(android.graphics.Bitmap)' on a null object reference
请添加
android {
...
aaptOptions {
noCompress "tflite"
}
}
- Code like this
val nsfwHelper = NSFWHelper.init(NSFWConfig(assets))
val nsfwBean = nsfwHelper?.scanBitmap(bitmap)!!
nsfwBean.sfw
nsfwBean.nsfw
if(nsfwBean.nsfw>0.3){
Log.e("NSFW","图片涉黄")
}