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

[mob] Bump minimum face score for clustering to 0.8

Browse source
This commit is contained in:
laurenspriem 2024-04-02 16:44:34 +05:30
parent e2ed836b16
commit 57fec06d73
3 changed files with 3 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
mobile/lib/services/machine_learning/face_ml/face_filtering/face_filtering_constants.dart
View file

@ -7,7 +7,7 @@ const kLaplacianThreshold = 15;
const kLapacianDefault = 10000.0;
/// The minimum score for a face to be considered a high quality face for clustering and person detection
const kMinHighQualityFaceScore = 0.78;
const kMinHighQualityFaceScore = 0.80;
/// The minimum score for a face to be detected, regardless of quality. Use [kMinHighQualityFaceScore] for high quality faces.
const kMinFaceDetectionScore = YoloOnnxFaceDetection.kMinScoreSigmoidThreshold;

2
mobile/lib/ui/settings/debug/face_debug_section_widget.dart
View file

@ -159,7 +159,7 @@ class _FaceDebugSectionWidgetState extends State<FaceDebugSectionWidget> {
trailingIconIsMuted: true,
onTap: () async {
await FaceMlService.instance
.clusterAllImages(minFaceScore: 0.75, clusterInBuckets: true);
.clusterAllImages(clusterInBuckets: true);
Bus.instance.fire(PeopleChangedEvent());
showShortToast(context, "Done");
},

2
mobile/lib/ui/viewer/file_details/faces_item_widget.dart
View file

@ -60,7 +60,7 @@ class FacesItemWidget extends StatelessWidget {
faces.sort((Face a, Face b) => b.score.compareTo(a.score));
// Remove faces with low scores and blurry faces
faces.removeWhere((face) => face.isHighQuality == false);
faces.removeWhere((face) => (face.isBlurry || face.score < 0.75));
// TODO: add deduplication of faces of same person
final faceIdsToClusterIds = await FaceMLDataDB.instance