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

MinorPerf: Avoid repeated sqrt computation

Browse source
This commit is contained in:
Neeraj Gupta 2024-01-09 15:32:43 +05:30
parent 5bbf227344
commit e21b862f53
1 changed files with 6 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
lib/services/semantic_search/frameworks/onnx/onnx_image_encoder.dart
View file

@ -43,9 +43,10 @@ class OnnxImageEncoder {
for (int y = 0; y < ny; y++) {
for (int x = 0; x < nx; x++) {
final int i = 3 * (y * nx + x);
inputImage[i] = rgb.getPixel(x, y).r.toDouble();
inputImage[i + 1] = rgb.getPixel(x, y).g.toDouble();
inputImage[i + 2] = rgb.getPixel(x, y).b.toDouble();
final pixel = rgb.getPixel(x, y);
inputImage[i] = pixel.r.toDouble();
inputImage[i + 1] = pixel.g.toDouble();
inputImage[i + 2] = pixel.b.toDouble();
}
}
@ -111,8 +112,9 @@ class OnnxImageEncoder {
for (int i = 0; i < 512; i++) {
imageNormalization += embedding[i] * embedding[i];
}
final double sqrtImageNormalization = sqrt(imageNormalization);
for (int i = 0; i < 512; i++) {
embedding[i] = embedding[i] / sqrt(imageNormalization);
embedding[i] = embedding[i] / sqrtImageNormalization;
}
return embedding;
}