浏览代码

Tag only objects with a score > 0.6

vishnukvmd 2 年之前
父节点
当前提交
32f554df84
共有 1 个文件被更改,包括 5 次插入1 次删除
  1. 5 1
      lib/services/object_detection/object_detection_service.dart

+ 5 - 1
lib/services/object_detection/object_detection_service.dart

@@ -8,6 +8,8 @@ import "package:photos/services/object_detection/tflite/classifier.dart";
 import "package:photos/services/object_detection/utils/isolate_utils.dart";
 
 class ObjectDetectionService {
+  static const scoreThreshold = 0.6;
+
   final _logger = Logger("ObjectDetectionService");
 
   /// Instance of [ObjectClassifier]
@@ -37,7 +39,9 @@ class ObjectDetectionService {
       final predictions = await _inference(isolateData);
       final Set<String> results = {};
       for (final Recognition result in predictions.recognitions) {
-        results.add(result.label);
+        if (result.score > scoreThreshold) {
+          results.add(result.label);
+        }
       }
       return results.toList();
     } catch (e, s) {