TerribleDev/ente
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

224 => 256

Browse Source 
https://github.com/apple/ml-mobileclip/blob/main/mobileclip/configs/mobileclip_s2.json
This commit is contained in:
Manav Rathi
2024-08-10 09:48:07 +05:30
parent 93c5825364
commit 5ce8d9838f
2 changed files with 2 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
desktop/src/main/services/ml-worker.ts
View File

@@ -220,7 +220,7 @@ const cachedCLIPImageSession = makeCachedInferenceSession(
export const computeCLIPImageEmbedding = async (input: Float32Array) => {
const session = await cachedCLIPImageSession();
const feeds = {
input: new ort.Tensor("float32", input, [1, 3, 224, 224]),
input: new ort.Tensor("float32", input, [1, 3, 256, 256]),
};
const t = Date.now();
const results = await session.run(feeds);

3
web/packages/new/photos/services/ml/clip.ts
View File

@@ -120,8 +120,7 @@ const computeEmbedding = async (
* Convert {@link imageData} into the format that the CLIP model expects.
*/
const convertToCLIPInput = (imageData: ImageData) => {
const requiredWidth = 224;
const requiredHeight = 224;
const [requiredWidth, requiredHeight] = [256, 256];
const mean = [0.48145466, 0.4578275, 0.40821073] as const;
const std = [0.26862954, 0.26130258, 0.27577711] as const;