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

[mob] Log time to fetch in debugMode

Browse source
This commit is contained in:
Neeraj Gupta 2024-03-28 13:40:15 +05:30
parent 03e5f06bf2
commit d6db3de6b1
1 changed files with 5 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
mobile/lib/services/face_ml/face_ml_service.dart
View file

@ -16,6 +16,7 @@ import "package:photos/core/event_bus.dart";
import "package:photos/db/ml_data_db.dart";
import "package:photos/events/diff_sync_complete_event.dart";
import "package:photos/extensions/list.dart";
import "package:photos/extensions/stop_watch.dart";
import "package:photos/face/db.dart";
import "package:photos/face/model/box.dart";
import "package:photos/face/model/detection.dart" as face_detection;
@ -458,8 +459,11 @@ class FaceMlService {
fileIds.add(f.uploadedFileID!);
}
try {
final EnteWatch? w = kDebugMode ? EnteWatch("face_em_fetch") : null;
w?.start();
final res =
await RemoteFileMLService.instance.getFilessEmbedding(fileIds);
w?.logAndReset('fetched ${res.mlData.length} embeddings');
final List<Face> faces = [];
final remoteFileIdToVersion = <int, int>{};
for (FileMl fileMl in res.mlData.values) {
@ -481,6 +485,7 @@ class FaceMlService {
remoteFileIdToVersion[fileMl.fileID] = fileMl.faceEmbedding.version;
}
await FaceMLDataDB.instance.bulkInsertFaces(faces);
w?.logAndReset('stored embeddings');
for (final entry in remoteFileIdToVersion.entries) {
alreadyIndexedFiles[entry.key] = entry.value;
}
@ -1167,7 +1172,6 @@ class FaceMlService {
if (!enteFile.isUploaded || enteFile.isOwner == false) {
return true;
}
// Skip if the file is a video
if (enteFile.fileType == FileType.video) {
return true;