修复demo bug

pull/34/head
赵文文 5 years ago
parent b0f57acad2
commit 10b2a1f300
  1. 7
      sample/src/main/java/com/zww/sample/model/AppRepository.kt

@ -3,6 +3,7 @@ package com.zww.sample.model
import android.app.Application import android.app.Application
import android.graphics.BitmapFactory import android.graphics.BitmapFactory
import android.os.Environment import android.os.Environment
import com.blankj.utilcode.util.ToastUtils
import com.google.gson.Gson import com.google.gson.Gson
import com.zww.sample.MyNSFWBean import com.zww.sample.MyNSFWBean
import com.zww.sample.ScoreBean import com.zww.sample.ScoreBean
@ -17,7 +18,6 @@ import java.io.File
import java.io.FileOutputStream import java.io.FileOutputStream
import java.io.InputStream import java.io.InputStream
import java.util.* import java.util.*
import kotlin.collections.ArrayList
/** /**
* 数据源 这里决定是从缓存获取还是从网络获取 * 数据源 这里决定是从缓存获取还是从网络获取
@ -31,7 +31,7 @@ class AppRepository constructor(
val mContext by inject(Application::class.java) val mContext by inject(Application::class.java)
val nsfwPath by lazy { val nsfwPath by lazy {
"${Environment.getExternalStorageDirectory().absolutePath}/nsfw.tflite" "/data/data/com.zww.sample/nsfw.tflite"
} }
//NSFW是否初始化完成 //NSFW是否初始化完成
@ -52,7 +52,7 @@ class AppRepository constructor(
val inputStream: InputStream = body.byteStream() val inputStream: InputStream = body.byteStream()
//创建一个文件夹 //创建一个文件夹
val file = File(Environment.getExternalStorageDirectory(), "nsfw.tflite") val file = File(nsfwPath)
val outputStream = FileOutputStream(file) val outputStream = FileOutputStream(file)
val bytes = ByteArray(1024) val bytes = ByteArray(1024)
var len = 0 var len = 0
@ -82,6 +82,7 @@ class AppRepository constructor(
*/ */
override fun initNSFW() { override fun initNSFW() {
Classifier.Build().context(mContext).isOpenGPU(true).nsfwModuleFilePath(nsfwPath).build() Classifier.Build().context(mContext).isOpenGPU(true).nsfwModuleFilePath(nsfwPath).build()
ToastUtils.showLong("NSFW初始化成功")
"NSFW初始化成功".d() "NSFW初始化成功".d()
isSDKInit = true isSDKInit = true
} }

Loading…
Cancel
Save