beenull/ente
1
0
Fork 0
Code Issues Pull requests Actions Packages Projects Releases Wiki Activity

[mob][photos] Check big and medium clusters first for suggestions

Browse source
This commit is contained in:
laurenspriem 2024-04-23 12:18:19 +05:30
parent 9e87b4a2cc
commit 6fe8dc7c66
1 changed files with 30 additions and 21 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

17
mobile/lib/services/machine_learning/face_ml/feedback/cluster_feedback.dart
View file

@ -1,5 +1,5 @@
import 'dart:developer' as dev;
import "dart:math" show Random;
import "dart:math" show Random, min;
import "package:flutter/foundation.dart";
import "package:logging/logging.dart";
@ -454,18 +454,25 @@ class ClusterFeedbackService {
name: "getSuggestionsUsingMedian",
);
// First only do a simple check on the big clusters
// First only do a simple check on the big clusters, if the person does not have small clusters yet
final EnteWatch watch = EnteWatch("ClusterFeedbackService")..start();
final smallestPersonClusterSize = personClusters
.map((clusterID) => allClusterIdsToCountMap[clusterID] ?? 0)
.reduce((value, element) => min(value, element));
final checkSizes = [kMinimumClusterSizeSearchResult, 20, 10, 5];
for (final minimumSize in checkSizes.toSet()) {
if (smallestPersonClusterSize >= minimumSize) {
final Map<int, List<double>> clusterAvgBigClusters =
await _getUpdateClusterAvg(
allClusterIdsToCountMap,
ignoredClusters,
minClusterSize: kMinimumClusterSizeSearchResult,
minClusterSize: minimumSize,
);
dev.log(
'computed avg for ${clusterAvgBigClusters.length} clusters, in ${DateTime.now().difference(startTime).inMilliseconds} ms',
);
final List<(int, double)> suggestionsMeanBigClusters = _calcSuggestionsMean(
final List<(int, double)> suggestionsMeanBigClusters =
_calcSuggestionsMean(
clusterAvgBigClusters,
personClusters,
ignoredClusters,
@ -476,6 +483,8 @@ class ClusterFeedbackService {
.map((e) => (e.$1, e.$2, true))
.toList(growable: false);
}
}
}
// Get and update the cluster summary to get the avg (centroid) and count
final Map<int, List<double>> clusterAvg = await _getUpdateClusterAvg(