首页、支付功能

app/src/main/java/com/sw/dualscreen/objbox/FoodModule.kt

@@ -10,6 +10,7 @@ import com.sw.dualscreen.MyApp
 import com.sw.dualscreen.utils.AssetsTool
 import com.sw.dualscreen.utils.GsonUtils
 import com.sw.dualscreen.utils.ImageUtil
+import com.sw.plate.App
 import io.objectbox.Box
 import io.objectbox.kotlin.boxFor
 import io.objectbox.query.Query
@@ -25,7 +26,7 @@ import java.io.InputStream
 object FoodModule {
 
     private const val THRESHOLD = 0.8
-    private lateinit var module_mobile: Module
+    private var module_mobile: Module?=null
     private lateinit var box: Box<Food>
     private lateinit var embeddingsList: List<List<Float>>
     private lateinit var labelsList: IntArray
@@ -61,7 +62,10 @@ object FoodModule {
             NO_MEAN_RGB, // [0.485, 0.456, 0.406] TORCHVISION_NORM_MEAN_RGB
             NO_STD_RGB    // [0.229, 0.224, 0.225] TORCHVISION_NORM_STD_RGB
         )
-        val outputTensor = module_mobile.forward(IValue.from(inputTensor)).toTensor()
+        if (module_mobile == null) {
+            init(App.getContext())
+        }
+        val outputTensor = module_mobile!!.forward(IValue.from(inputTensor)).toTensor()
         return outputTensor.dataAsFloatArray
     }
 
@@ -103,6 +107,9 @@ object FoodModule {
     fun getFoodScoreList(bitmap: Bitmap, queryCount: Int = 15): List<IdNameScore> {
         val floatArray = bitmap2FloatArray(bitmap)
         val nameScoreList = queryFoodNameScore(floatArray, queryCount)
+        if (nameScoreList.isEmpty()) {
+            return emptyList()
+        }
         val maxScoreList = nameScoreList
             .filter { it.score < 1 - THRESHOLD }
             .groupBy { it.name }