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

Fix local env needing refresh to login

Browse source 
by loading tfjs-tflite as dynamic import
This commit is contained in:
Shailesh Pandit 2022-01-16 13:01:29 +05:30
parent 18db6f44dc
commit 43c792e46d
2 changed files with 7 additions and 2043 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2039
public/js/tflite/tflite_web_api_client.js
View file

File diff suppressed because it is too large Load diff

11
src/services/machineLearning/mobileFaceNetEmbeddingService.ts
View file

@ -1,5 +1,5 @@
import * as tf from '@tensorflow/tfjs-core'; import * as tf from '@tensorflow/tfjs-core';
import * as tflite from '@tensorflow/tfjs-tflite'; import { TFLiteModel } from '@tensorflow/tfjs-tflite';
import PQueue from 'p-queue'; import PQueue from 'p-queue';
import { import {
FaceEmbedding, FaceEmbedding,
@ -14,11 +14,10 @@ class MobileFaceNetEmbeddingService implements FaceEmbeddingService {
public method: Versioned<FaceEmbeddingMethod>; public method: Versioned<FaceEmbeddingMethod>;
public faceSize: number; public faceSize: number;
private mobileFaceNetModel: Promise<tflite.TFLiteModel>; private mobileFaceNetModel: Promise<TFLiteModel>;
private serialQueue: PQueue; private serialQueue: PQueue;
public constructor(faceSize: number = MOBILEFACENET_FACE_SIZE) { public constructor(faceSize: number = MOBILEFACENET_FACE_SIZE) {
tflite.setWasmPath('/js/tflite/');
this.method = { this.method = {
value: 'MobileFaceNet', value: 'MobileFaceNet',
version: 2, version: 2,
@ -29,6 +28,10 @@ class MobileFaceNetEmbeddingService implements FaceEmbeddingService {
} }
private async init() { private async init() {
// TODO: can also create new instance per new syncContext
const tflite = await import('@tensorflow/tfjs-tflite');
tflite.setWasmPath('/js/tflite/');
this.mobileFaceNetModel = tflite.loadTFLiteModel( this.mobileFaceNetModel = tflite.loadTFLiteModel(
'/models/mobilefacenet/mobilefacenet.tflite' '/models/mobilefacenet/mobilefacenet.tflite'
); );
@ -50,7 +53,7 @@ class MobileFaceNetEmbeddingService implements FaceEmbeddingService {
public getFaceEmbeddingTF( public getFaceEmbeddingTF(
faceTensor: tf.Tensor4D, faceTensor: tf.Tensor4D,
mobileFaceNetModel: tflite.TFLiteModel mobileFaceNetModel: TFLiteModel
): tf.Tensor2D { ): tf.Tensor2D {
return tf.tidy(() => { return tf.tidy(() => {
const normalizedFace = tf.sub(tf.div(faceTensor, 127.5), 1.0); const normalizedFace = tf.sub(tf.div(faceTensor, 127.5), 1.0);