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

[mob] Minor changes to clustering

Browse source
This commit is contained in:
laurenspriem 2024-03-21 18:07:12 +05:30
parent b5cff212bb
commit f94aa400bf
2 changed files with 8 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
mobile/lib/services/machine_learning/face_ml/face_clustering/linear_clustering_service.dart
View file

@ -131,12 +131,13 @@ class FaceLinearClustering {
final errorStackTrace = receivedMessage['stackTrace'];
final exception = Exception(errorMessage);
final stackTrace = StackTrace.fromString(errorStackTrace);
_activeTasks--;
completer.completeError(exception, stackTrace);
} else {
_activeTasks--;
completer.complete(receivedMessage);
}
});
_activeTasks--;
return completer.future;
}

7
mobile/lib/services/machine_learning/face_ml/face_ml_service.dart
View file

@ -374,11 +374,16 @@ class FaceMlService {
final faceIdToEmbedding = await FaceMLDataDB.instance.getFaceEmbeddingMap(
minScore: minFaceScore,
);
_logger.info('read embeddings ${faceIdToEmbedding.length} ');
final gotFaceEmbeddingsTime = DateTime.now();
_logger.info(
'read embeddings ${faceIdToEmbedding.length} in ${gotFaceEmbeddingsTime.difference(clusterStartTime).inMilliseconds} ms',
);
// Read the creation times from Files DB, in a map from fileID to creation time
final fileIDToCreationTime =
await FilesDB.instance.getFileIDToCreationTime();
_logger.info('read creation times from FilesDB in '
'${DateTime.now().difference(gotFaceEmbeddingsTime).inMilliseconds} ms');
// Cluster the embeddings using the linear clustering algorithm, returning a map from faceID to clusterID
final faceIdToCluster = await FaceLinearClustering.instance.predict(