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

Inline

Browse source
This commit is contained in:
Manav Rathi 2024-05-18 10:56:19 +05:30
parent 4840214d89
commit e13f59606f
No known key found for this signature in database
2 changed files with 5 additions and 12 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -24,7 +24,6 @@ import {
fetchImageBitmap,
getLocalFileImageBitmap,
getThumbnailImageBitmap,
normalizePixelBetween0And1,
pixelRGBBilinear,
warpAffineFloat32List,
} from "./image";
@ -194,7 +193,7 @@ const convertToYOLOInputFloat32ChannelsFirst = (imageBitmap: ImageBitmap) => {
const yoloSize = { width: scaledWidth, height: scaledHeight };
// Populate the Float32Array with normalized pixel values.
let pixelIndex = 0;
let pi = 0;
const channelOffsetGreen = requiredHeight * requiredWidth;
const channelOffsetBlue = 2 * requiredHeight * requiredWidth;
for (let h = 0; h < requiredHeight; h++) {
@ -209,12 +208,10 @@ const convertToYOLOInputFloat32ChannelsFirst = (imageBitmap: ImageBitmap) => {
width,
height,
);
yoloInput[pixelIndex] = normalizePixelBetween0And1(r);
yoloInput[pixelIndex + channelOffsetGreen] =
normalizePixelBetween0And1(g);
yoloInput[pixelIndex + channelOffsetBlue] =
normalizePixelBetween0And1(b);
pixelIndex++;
yoloInput[pi] = r / 255.0;
yoloInput[pi + channelOffsetGreen] = g / 255.0;
yoloInput[pi + channelOffsetBlue] = b / 255.0;
pi++;
}
}

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

@ -52,10 +52,6 @@ export async function getLocalFileImageBitmap(
return createImageBitmap(fileBlob);
}
export function normalizePixelBetween0And1(pixelValue: number) {
return pixelValue / 255.0;
}
export function normalizePixelBetweenMinus1And1(pixelValue: number) {
return pixelValue / 127.5 - 1.0;
}