You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
Jason e6d8dd6bd3
Create LICENSE
5 years ago
.idea 升级gradle与部分第三方库 6 years ago
app Upgrade demo version 6 years ago
gradle/wrapper 升级gradle与部分第三方库 6 years ago
img Upgrade demo version 6 years ago
nsfw 更新Kotlin拓展函数支持,调用扫描更方便。 6 years ago
.gitattributes Update .gitattributes 6 years ago
.gitignore Match GitHub usage protocol 6 years ago
LICENSE Create LICENSE 5 years ago
README.md Update README.md 6 years ago
build.gradle 升级gradle与部分第三方库 6 years ago
gradle.properties 更新初始化方式,初始化时默认关闭GPU加速防止部分手机奔溃 6 years ago
gradlew first commit. 6 years ago
gradlew.bat first commit. 6 years ago
settings.gradle 移除采样过程中的水平旋转、移除opencv库,修复:https://github.com/devzwy/open_nsfw_android/issues/4#issuecomment-494745262 6 years ago

README.md

open_nsfw_android

Mozilla Add-on

色情图片离线识别,基于TensorFlow实现。识别只需200ms,可断网测试,成功率99%,调用只要一行代码,从雅虎的开源项目open_nsfw移植,tflite(6M)为训练好的模型(已量化),该模型文件可用于iOS、java、C++等平台,Python使用生成的tfLite文件检测图片的速度远远快于使用原模型.

iOS请参考:issues13

Python参考:NSFW-PythonPython-TensorflowLite-ApiPython-Tensorflow-Api

Java参考:Tensorflow-Api

C++参考:TensorflowLite-Api

JavaScript参考JS相关文档

其中Python、C++均有两种数据喂入的方式,可根据需求选择pb模型或tfLite文件,详细请参考上面的链接.Java的目前只能加载pb模型。其他的平台可自行百度

pb模型下载

本项目移除测试图片,请下载Demo后自行配图测试

使用

  • Add it in your root build.gradle at the end of repositories:
	allprojects {
		repositories {
			...
			maven { url 'https://jitpack.io' }
		}
	}
  • Add the dependency
	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"
        }
  }
  • 使用
  val nsfwHelper = NSFWHelper.init(NSFWConfig(assets))
  val nsfwBean = nsfwHelper?.scanBitmap(bitmap)!!
  nsfwBean.sfw
  nsfwBean.nsfw
  if(nsfwBean.nsfw>0.3){
    Log.e("NSFW","图片涉黄")
  }
  • kotlin可直接使用File.getNsfwScore(mAssetManager: AssetManager): NsfwBean 或 Bitmap.getNsfwScore(mAssetManager: AssetManager): NsfwBean 直接获取鉴定结果(NSFWHelper 1.2.9版本开始支持),比如:
  val bitmap = BitmapFactory.decodeFile(path)
  
  val nsfwScore = bitmap.getNsfwScore(assets)
  
  if(nsfwBean.nsfw>0.3){
  
      Log.e("NSFW","图片涉黄")
      
  }

      val file = File(lm.path)
      
      val nsfwScore = file.getNsfwScore(assets)
      
      if(nsfwBean.nsfw>0.3){
      
          Log.e("NSFW","图片涉黄")
          
      }
      

安卓手机直接点我安装

扫码下载

图片

Demo运行结果:

图片