Manav Rathi 1 year ago
parent
commit
cfbd7806c8

+ 2 - 5
web/apps/photos/src/services/machineLearning/faceService.ts

@@ -245,12 +245,9 @@ class FaceService {
     ) {
         // await this.init();
 
-        const clusteringConfig = DEFAULT_ML_SYNC_CONFIG.faceClustering;
-
-        if (!allFaces || allFaces.length < clusteringConfig.minInputSize) {
+        if (!allFaces || allFaces.length < 50) {
             log.info(
-                "[MLService] Too few faces to cluster, not running clustering: ",
-                allFaces.length,
+                `Skipping clustering since number of faces (${allFaces.length}) is less than the clustering threshold (50)`,
             );
             return;
         }

+ 5 - 5
web/apps/photos/src/services/machineLearning/hdbscanClusteringService.ts

@@ -26,11 +26,11 @@ class HdbscanClusteringService implements ClusteringService {
         const hdbscan = new Hdbscan({
             input,
 
-            minClusterSize: config.minClusterSize,
-            minSamples: config.minSamples,
-            clusterSelectionEpsilon: config.clusterSelectionEpsilon,
-            clusterSelectionMethod: config.clusterSelectionMethod,
-            debug: config.generateDebugInfo,
+            minClusterSize: 3,
+            minSamples: 5,
+            clusterSelectionEpsilon: 0.6,
+            clusterSelectionMethod: "leaf",
+            debug: true,
         });
 
         return {