From 0329c40265f7eace8a7a2d8cd82cb083513e9119 Mon Sep 17 00:00:00 2001 From: Manav Rathi Date: Tue, 3 Sep 2024 17:36:21 +0530 Subject: [PATCH] Tweak --- web/apps/photos/src/pages/cluster-debug.tsx | 4 ++-- web/packages/new/photos/services/ml/cluster.ts | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/web/apps/photos/src/pages/cluster-debug.tsx b/web/apps/photos/src/pages/cluster-debug.tsx index 50e0d5da56..3b9a7ad230 100644 --- a/web/apps/photos/src/pages/cluster-debug.tsx +++ b/web/apps/photos/src/pages/cluster-debug.tsx @@ -11,7 +11,7 @@ import { type FaceF32, type OnClusteringProgress, } from "@/new/photos/services/ml/cluster"; -import { faceDirection, type Face } from "@/new/photos/services/ml/face"; +import { faceDirection } from "@/new/photos/services/ml/face"; import type { EnteFile } from "@/new/photos/types/file"; import { FlexWrapper, @@ -68,7 +68,7 @@ export default function ClusterDebug() { minBlur: 10, minScore: 0.8, minClusterSize: 2, - joinThreshold: 0.7, + joinThreshold: 0.6, earlyExitThreshold: 0.9, batchSize: 10000, offsetIncrement: 7500, diff --git a/web/packages/new/photos/services/ml/cluster.ts b/web/packages/new/photos/services/ml/cluster.ts index 282a6c2439..1aa665666a 100644 --- a/web/packages/new/photos/services/ml/cluster.ts +++ b/web/packages/new/photos/services/ml/cluster.ts @@ -250,7 +250,7 @@ export const clusterFaces = ( for (let offset = 0; offset < faces.length; offset += offsetIncrement) { const it = Date.now(); - const batch = faces.slice(offset, batchSize); + const batch = faces.slice(offset, offset + batchSize); log.info(`[batch] processing ${offset} to ${offset + batch.length}`); const oldState = {