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

[mob] Fix too many clustering rounds/buckets

Browse source
This commit is contained in:
laurenspriem 2024-04-13 15:54:36 +05:30
parent 8801dc1a7a
commit 2ce078e173
1 changed files with 7 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -390,6 +390,9 @@ class FaceMlService {
batchSize: batchSize,
);
if (faceIdToEmbeddingBucket.isEmpty) {
_logger.warning(
'faceIdToEmbeddingBucket is empty, this should ideally not happen as it should have stopped earlier. offset: $offset, totalFaces: $totalFaces',
);
break;
}
if (offset > totalFaces) {
@ -414,6 +417,10 @@ class FaceMlService {
_logger.info(
'Done with clustering ${offset + faceIdToEmbeddingBucket.length} embeddings (${(100 * (offset + faceIdToEmbeddingBucket.length) / totalFaces).toStringAsFixed(0)}%) in bucket $bucket, offset: $offset',
);
if (offset + bucketSize >= totalFaces) {
_logger.info('All faces clustered');
break;
}
offset += offsetIncrement;
bucket++;
}