[mob][photos] Clip logs

This commit is contained in:
laurenspriem 2024-08-26 16:15:49 +02:00
parent 1f05ae268f
commit e98e8833e6
2 changed files with 7 additions and 1 deletions

View File

@ -55,7 +55,9 @@ class ClipImageEncoder extends MlModel {
final embedding = (outputs[0]?.value as List<List<double>>)[0];
inputOrt.release();
runOptions.release();
outputs.forEach((element) => element?.release());
for (var element in outputs) {
element?.release();
}
normalizeEmbedding(embedding);
w.stopWithLog("done");
return embedding;

View File

@ -303,6 +303,7 @@ class SemanticSearchService {
ByteData imageByteData,
int clipImageAddress,
) async {
final startTime = DateTime.now();
final embedding = await ClipImageEncoder.predict(
image,
imageByteData,
@ -311,6 +312,9 @@ class SemanticSearchService {
final clipResult = ClipResult(fileID: enteFileID, embedding: embedding);
dev.log('Finished running ClipImage for $enteFileID in '
'${DateTime.now().difference(startTime).inMilliseconds} ms');
return clipResult;
}
}