Переглянути джерело

Tag only objects with a score > 0.6

vishnukvmd 2 роки тому
батько
коміт
32f554df84

+ 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) {