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

Trim

Browse source
This commit is contained in:
Manav Rathi 2024-05-18 15:53:10 +05:30
parent 996d9ccda5
commit bef7574c29
No known key found for this signature in database
1 changed files with 8 additions and 12 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -38,18 +38,18 @@ import { transformFaceDetections } from "./transform-box";
* 1. Downloading the original if needed. * 1. Downloading the original if needed.
* 2. Detect faces using ONNX/YOLO * 2. Detect faces using ONNX/YOLO
* 3. Align the face rectangles, compute blur. * 3. Align the face rectangles, compute blur.
* 4. Compute embbeddings for the detected face (crops). * 4. Compute embeddings for the detected face (crops).
* *
* Once all of it is done, it returns the face rectangles and embeddings to the * Once all of it is done, it returns the face rectangles and embeddings so that
* higher layer (which saves them to locally for offline use, and encrypts and * they can be saved locally for offline use, and encrypts and uploads them to
* uploads them to the user's remote storage so that their other devices can * the user's remote storage so that their other devices can download them
* download them instead of needing to reindex). * instead of needing to reindex.
*/ */
export const indexFaces = async ( export const indexFaces = async (
enteFile: EnteFile, enteFile: EnteFile,
localFile?: globalThis.File, localFile?: globalThis.File,
) => { ) => {
log.debug(() => ({ a: "Indexing faces in file", enteFile })); const startTime = Date.now();
const fileContext: MLSyncFileContext = { enteFile, localFile }; const fileContext: MLSyncFileContext = { enteFile, localFile };
const newMlFile = (fileContext.newMlFile = { const newMlFile = (fileContext.newMlFile = {
@ -66,6 +66,8 @@ export const indexFaces = async (
fileContext.imageBitmap && fileContext.imageBitmap.close(); fileContext.imageBitmap && fileContext.imageBitmap.close();
} }
const ms = Math.round(Date.now() - startTime);
log.debug(() => `Indexing faces in file ${enteFile.id} took ${ms} ms`);
return newMlFile; return newMlFile;
}; };
@ -103,8 +105,6 @@ const fetchImageBitmapForContext = async (fileContext: MLSyncFileContext) => {
const syncFileAnalyzeFaces = async (fileContext: MLSyncFileContext) => { const syncFileAnalyzeFaces = async (fileContext: MLSyncFileContext) => {
const { newMlFile } = fileContext; const { newMlFile } = fileContext;
const startTime = Date.now();
await syncFileFaceDetections(fileContext); await syncFileFaceDetections(fileContext);
if (newMlFile.faces && newMlFile.faces.length > 0) { if (newMlFile.faces && newMlFile.faces.length > 0) {
@ -116,10 +116,6 @@ const syncFileAnalyzeFaces = async (fileContext: MLSyncFileContext) => {
await syncFileFaceMakeRelativeDetections(fileContext); await syncFileFaceMakeRelativeDetections(fileContext);
} }
log.debug(
() =>
`Face detection for file ${fileContext.enteFile.id} took ${Math.round(Date.now() - startTime)} ms`,
);
}; };
const syncFileFaceDetections = async (fileContext: MLSyncFileContext) => { const syncFileFaceDetections = async (fileContext: MLSyncFileContext) => {