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

Compute rejected clusters

Browse Source
This commit is contained in:
Manav Rathi
2024-11-16 09:27:20 +05:30
parent 8cf87acb7b
commit 0e6705f8e1
1 changed files with 20 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
web/packages/new/photos/services/ml/cluster.ts
View File

@@ -97,6 +97,26 @@ export const _clusterFaces = async (
const sortedCGroups = cgroups.sort((a, b) => b.updatedAt - a.updatedAt);
const rejectedClusterIDsForFaceID = new Map<string, Set<string>>();
for (const cgroup of sortedCGroups) {
if (cgroup.data.rejectedFaceIDs.length == 0) {
clusters = clusters.concat(cgroup.data.assigned);
} else {
const rejectedFaceIDs = new Set(cgroup.data.rejectedFaceIDs);
clusters = clusters.concat(
cgroup.data.assigned.map((cluster) => ({
...cluster,
faces: cluster.faces.filter((f) => !rejectedFaceIDs.has(f)),
})),
);
for (const faceID of rejectedFaceIDs) {
const s = rejectedClusterIDsForFaceID.get(faceID) ?? new Set();
cgroup.data.assigned.forEach(({ id }) => s.add(id));
rejectedClusterIDsForFaceID.set(faceID, s);
}
}
}
// Extract the remote clusters.
clusters = clusters.concat(
// See: [Note: strict mode migration]