TerribleDev/ente
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

[mob][photos] Assign not person feedback for completely rejected remote cluster

Browse Source
This commit is contained in:
laurenspriem
2024-11-12 14:54:25 +05:30
parent 015790874e
commit b9c63426fc
1 changed files with 17 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
mobile/lib/services/machine_learning/face_ml/person/person_service.dart
View File

@@ -319,19 +319,20 @@ class PersonService {
logger.info(
"Person ${e.id} ${personData.name} has ${remotelyRejectedFaceIDs.length} rejected faces",
);
// Assign rejected faces to new clusters
for (final faceId in remotelyRejectedFaceIDs) {
faceIdToClusterID[faceId] = newClusterID();
}
// Check that we don't have any empty clusters now
final dbPersonClusterInfo =
await faceMLDataDB.getClusterIdToFaceIdsForPerson(e.id);
for (final clusterIdToFaceIDs in dbPersonClusterInfo.entries) {
final clusterID = clusterIdToFaceIDs.key;
final faceIDs = clusterIdToFaceIDs.value;
faceIDs.removeWhere(
(faceID) => remotelyRejectedFaceIDs.contains(faceID),
);
final foundRejectedFaces = <String>[];
for (final faceID in faceIDs) {
if (remotelyRejectedFaceIDs.contains(faceID)) {
faceIDs.remove(faceID);
foundRejectedFaces.add(faceID);
}
}
if (faceIDs.isEmpty) {
logger.info(
"Cluster $clusterID for person ${e.id} ${personData.name} is empty due to rejected faces from remote, removing the cluster from person",
@@ -340,8 +341,17 @@ class PersonService {
personID: e.id,
clusterID: clusterID,
);
await faceMLDataDB.captureNotPersonFeedback(
personID: e.id,
clusterID: clusterID,
);
remotelyRejectedFaceIDs.removeAll(foundRejectedFaces);
}
}
// Assign rejected faces to new clusters
for (final faceId in remotelyRejectedFaceIDs) {
faceIdToClusterID[faceId] = newClusterID();
}
}
}
}