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