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

Fix duplicate check

Browse source
This commit is contained in:
Manav Rathi 2024-05-18 20:50:35 +05:30
parent 93c498b0f4
commit 3ade7b797e
No known key found for this signature in database
1 changed files with 5 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
web/apps/photos/src/services/face/f-index.ts
View file

@ -293,13 +293,12 @@ const removeDuplicateDetections = (
) => { ) => {
detections.sort((a, b) => b.probability - a.probability); detections.sort((a, b) => b.probability - a.probability);
const isSelected = new Map<number, boolean>(); const dupIndices = new Set<number>();
for (let i = 0; i < detections.length; i++) { for (let i = 0; i < detections.length; i++) {
if (!isSelected.get(i)) continue; if (dupIndices.has(i)) continue;
isSelected.set(i, true);
for (let j = i + 1; j < detections.length; j++) { for (let j = i + 1; j < detections.length; j++) {
if (!isSelected.get(j)) continue; if (dupIndices.has(j)) continue;
const centeri = faceDetectionCenter(detections[i]); const centeri = faceDetectionCenter(detections[i]);
const centerj = faceDetectionCenter(detections[j]); const centerj = faceDetectionCenter(detections[j]);
@ -308,11 +307,11 @@ const removeDuplicateDetections = (
[centerj.x, centerj.y], [centerj.x, centerj.y],
); );
if (dist <= withinDistance) isSelected.set(j, false); if (dist <= withinDistance) dupIndices.add(j);
} }
} }
return detections.filter((_, i) => isSelected.get(i)); return detections.filter((_, i) => !dupIndices.has(i));
}; };
const faceDetectionCenter = (detection: FaceDetection) => { const faceDetectionCenter = (detection: FaceDetection) => {