From 4d8aedbaa7f4b55e97a541213f65f6910121dc1c Mon Sep 17 00:00:00 2001 From: lvmeng <848755140@qq.com> Date: Tue, 4 Nov 2025 13:53:35 +0800 Subject: [PATCH] =?UTF-8?q?=E5=A2=9E=E5=8A=A0=E8=AF=86=E5=88=AB=E6=97=A5?= =?UTF-8?q?=E5=BF=97?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../main/java/com/sw/inbound/objbox/FoodModule.kt | 15 ++++++++++++++- 1 file changed, 14 insertions(+), 1 deletion(-) diff --git a/app/src/main/java/com/sw/inbound/objbox/FoodModule.kt b/app/src/main/java/com/sw/inbound/objbox/FoodModule.kt index 05edd28..9b746d0 100644 --- a/app/src/main/java/com/sw/inbound/objbox/FoodModule.kt +++ b/app/src/main/java/com/sw/inbound/objbox/FoodModule.kt @@ -8,12 +8,14 @@ import com.google.gson.reflect.TypeToken import com.sw.inbound.MyApp import com.sw.inbound.utils.AssetsTool import com.sw.inbound.utils.ImageUtil +import com.sw.inbound.utils.ext.toJsonString import io.objectbox.Box import io.objectbox.kotlin.boxFor import io.objectbox.query.Query import org.pytorch.IValue import org.pytorch.Module import org.pytorch.torchvision.TensorImageUtils +import timber.log.Timber import java.io.File import java.io.FileOutputStream import java.io.IOException @@ -88,7 +90,13 @@ object FoodModule { //查询比较分数 // val tempList = query.findWithScores().sortedBy { it.score }.map { "${it.get().name}|${it.get().foodIdx}|${it.score}" } val map = mutableMapOf() - query.findIdsWithScores().filter { it.score < 0.2 }.forEach { + val idScoreList = query.findIdsWithScores() + val nameScoreList = mutableListOf() + idScoreList.forEach { + nameScoreList.add(IdNameScore(id = it.id, name = box.get(it.id).name?:"", score = it.score)) + } + Timber.tag("FoodModule").d("queryFood数据:${nameScoreList.toJsonString()}") + idScoreList.filter { it.score < 0.2 }.forEach { val food = box.get(it.id) //FoodQueryResult(id = it.id, name = food.name, foodIdx = food.foodIdx, score = it.score) food.name?.let { key -> @@ -100,6 +108,11 @@ object FoodModule { return list } + data class IdNameScore( + val id:Long, + val name:String, + val score: Double + ) fun initDefFoodData(context: Context, action:()-> Unit={}) { val count = box.all.count { it.foodIdx == DEFAULT_FOOD_INDEX }