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

Reduce distance to server type

Browse source
This commit is contained in:
Manav Rathi 2024-05-19 13:23:08 +05:30
parent 6b5788539b
commit ee894a668c
No known key found for this signature in database
2 changed files with 6 additions and 14 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
web/apps/photos/src/services/face/types.ts
View file

@ -2,12 +2,10 @@ import { Box, Dimensions, Point } from "services/face/geom";
export declare type Cluster = Array<number>;
export declare type Landmark = Point;
export interface FaceDetection {
// box and landmarks is relative to image dimentions stored at mlFileData
box: Box;
landmarks?: Array<Landmark>;
landmarks?: Point[];
probability?: number;
}

16
web/apps/photos/src/services/machineLearning/machineLearningService.ts
View file

@ -10,7 +10,8 @@ import PQueue from "p-queue";
import { putEmbedding } from "services/embeddingService";
import mlIDbStorage, { ML_SEARCH_CONFIG_NAME } from "services/face/db";
import { fetchImageBitmap, getLocalFile } from "services/face/file";
import { Face, FaceDetection, Landmark, MlFileData } from "services/face/types";
import type { Point } from "services/face/geom";
import { Face, FaceDetection, MlFileData } from "services/face/types";
import { getLocalFiles } from "services/fileService";
import { EnteFile } from "types/file";
import { isInternalUserForML } from "utils/user";
@ -483,9 +484,9 @@ class ServerFace {
class ServerDetection {
public box: ServerFaceBox;
public landmarks: Landmark[];
public landmarks: Point[];
public constructor(box: ServerFaceBox, landmarks: Landmark[]) {
public constructor(box: ServerFaceBox, landmarks: Point[]) {
this.box = box;
this.landmarks = landmarks;
}
@ -529,19 +530,12 @@ function LocalFileMlDataToServerFileMl(
const box = detection.box;
const landmarks = detection.landmarks;
const newBox = new ServerFaceBox(box.x, box.y, box.width, box.height);
const newLandmarks: Landmark[] = [];
for (let j = 0; j < landmarks.length; j++) {
newLandmarks.push({
x: landmarks[j].x,
y: landmarks[j].y,
} as Landmark);
}
// TODO-ML: Add client UA and version
const newFaceObject = new ServerFace(
faceID,
Array.from(embedding),
new ServerDetection(newBox, newLandmarks),
new ServerDetection(newBox, landmarks),
score,
blur,
);