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

[mob][photos] Increase conservative clustering threshold slightly

Browse source
This commit is contained in:
laurenspriem 2024-04-25 16:41:23 +05:30
parent f101468a8d
commit 3828fa328e
1 changed files with 4 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
mobile/lib/services/machine_learning/face_ml/face_clustering/face_clustering_service.dart
View file

@ -69,7 +69,7 @@ class FaceClusteringService {
bool isRunning = false;
static const kRecommendedDistanceThreshold = 0.24;
static const kConservativeDistanceThreshold = 0.15;
static const kConservativeDistanceThreshold = 0.16;
// singleton pattern
FaceClusteringService._privateConstructor();
@ -961,9 +961,9 @@ class FaceClusteringService {
// Run the DBSCAN clustering
final List<List<int>> clusterOutput = dbscan.run(embeddings);
final List<List<FaceInfo>> clusteredFaceInfos = clusterOutput
.map((cluster) => cluster.map((idx) => faceInfos[idx]).toList())
.toList();
// final List<List<FaceInfo>> clusteredFaceInfos = clusterOutput
// .map((cluster) => cluster.map((idx) => faceInfos[idx]).toList())
// .toList();
final List<List<String>> clusteredFaceIDs = clusterOutput
.map((cluster) => cluster.map((idx) => faceInfos[idx].faceID).toList())
.toList();