beenull/ente
1
0
Fork 0
Code Issues Pull requests Actions Packages Projects Releases Wiki Activity

Inline

Browse source
This commit is contained in:
Manav Rathi 2024-05-16 11:10:41 +05:30
parent 2c0e8c76c3
commit cfbd7806c8
No known key found for this signature in database
2 changed files with 7 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
web/apps/photos/src/services/machineLearning/faceService.ts
View file

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

10
web/apps/photos/src/services/machineLearning/hdbscanClusteringService.ts
View file

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