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

[mob] Use linear clustring for breaking up clusters

Browse source
This commit is contained in:
Neeraj Gupta 2024-04-08 08:52:00 +05:30
parent 412e93c9fa
commit 0b08afdbe2
1 changed files with 9 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
mobile/lib/services/machine_learning/face_ml/feedback/cluster_feedback.dart
View file

@ -408,8 +408,10 @@ class ClusterFeedbackService {
// TODO: iterate over this method to find sweet spot
Future<Map<int, List<String>>> breakUpCluster(
int clusterID, {
useDbscan = true,
useDbscan = false,
}) async {
_logger.info(
'breakUpCluster called for cluster $clusterID with dbscan $useDbscan');
final faceMlDb = FaceMLDataDB.instance;
final faceIDs = await faceMlDb.getFaceIDsForCluster(clusterID);
@ -450,11 +452,16 @@ class ClusterFeedbackService {
final faceIdToCluster = await FaceClustering.instance.predictLinear(
clusteringInput,
fileIDToCreationTime: fileIDToCreationTime,
distanceThreshold: 0.15,
distanceThreshold: 0.275,
);
if (faceIdToCluster == null) {
_logger.info('No clusters found');
return {};
} else {
_logger.info(
'Broke up cluster $clusterID into ${faceIdToCluster.values.toSet().length} clusters',
);
}
for (final entry in faceIdToCluster.entries) {