From 3bc41f253b25c7bdb51bd9388b8592a0014d11c1 Mon Sep 17 00:00:00 2001 From: Manav Rathi Date: Fri, 16 Aug 2024 15:22:02 +0530 Subject: [PATCH] Propagate --- .../new/photos/services/ml/cluster-new.ts | 10 +++++----- web/packages/new/photos/services/ml/index.ts | 16 ++++++++-------- 2 files changed, 13 insertions(+), 13 deletions(-) diff --git a/web/packages/new/photos/services/ml/cluster-new.ts b/web/packages/new/photos/services/ml/cluster-new.ts index 69e8de6f26..ff28b75260 100644 --- a/web/packages/new/photos/services/ml/cluster-new.ts +++ b/web/packages/new/photos/services/ml/cluster-new.ts @@ -241,10 +241,10 @@ export const clusterFaces = async (faceIndexes: FaceIndex[]) => { // clusters as its display face. const faceForFaceID = new Map(faces.map((f) => [f.faceID, f])); - const people = await clusterGroups(); + const cgroups = await clusterGroups(); - for (const person of people) { - person.avatarFaceID = person.clusterIDs + for (const cgroup of cgroups) { + cgroup.avatarFaceID = cgroup.clusterIDs .map((clusterID) => clusterIndexForClusterID.get(clusterID)) .map((clusterIndex) => clusterIndex ? clusters[clusterIndex] : undefined, @@ -265,7 +265,7 @@ export const clusterFaces = async (faceIndexes: FaceIndex[]) => { validClusters, clusterIndexForClusterID, clusterIDForFaceID, - people, + cgroups, }, ]); log.debug( @@ -273,7 +273,7 @@ export const clusterFaces = async (faceIndexes: FaceIndex[]) => { `Clustered ${faces.length} faces into ${validClusters.length} clusters (${Date.now() - t} ms)`, ); - return { clusters: validClusters, people }; + return { clusters: validClusters, cgroups }; }; /** diff --git a/web/packages/new/photos/services/ml/index.ts b/web/packages/new/photos/services/ml/index.ts index 65a36c5981..670bc5cc58 100644 --- a/web/packages/new/photos/services/ml/index.ts +++ b/web/packages/new/photos/services/ml/index.ts @@ -347,7 +347,7 @@ export const wipCluster = async () => { if (last) return last; - const { clusters, people } = await clusterFaces(await faceIndexes()); + const { clusters, cgroups } = await clusterFaces(await faceIndexes()); const clusterByID = new Map( clusters.map((cluster) => [cluster.id, cluster]), ); @@ -356,31 +356,31 @@ export const wipCluster = async () => { const localFilesByID = new Map(localFiles.map((f) => [f.id, f])); const result: SearchPerson[] = []; - for (const person of people) { - let avatarFaceID = person.avatarFaceID; + for (const cgroup of cgroups) { + let avatarFaceID = cgroup.avatarFaceID; // TODO-Cluster // Temp if (!avatarFaceID) { // eslint-disable-next-line @typescript-eslint/no-non-null-assertion - avatarFaceID = person.clusterIDs + avatarFaceID = cgroup.clusterIDs .map((id) => clusterByID.get(id)) .flatMap((cluster) => cluster?.faceIDs ?? [])[0]!; } - person.clusterIDs; + cgroup.clusterIDs; const avatarFaceFileID = fileIDFromFaceID(avatarFaceID); const avatarFaceFile = localFilesByID.get(avatarFaceFileID ?? 0); if (!avatarFaceFileID || !avatarFaceFile) { assertionFailed(`Face ID ${avatarFaceID} without local file`); continue; } - const files = person.clusterIDs + const files = cgroup.clusterIDs .map((id) => clusterByID.get(id)) .flatMap((cluster) => cluster?.faceIDs ?? []) .map((faceID) => fileIDFromFaceID(faceID)) .filter((fileID) => fileID !== undefined); result.push({ - id: person.id, - name: person.name, + id: cgroup.id, + name: cgroup.name, files, displayFaceID: avatarFaceID, displayFaceFile: avatarFaceFile,