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

Run tensorflow.js only on webgl

Browse source
This commit is contained in:
Shailesh Pandit 2022-01-30 11:03:02 +05:30
parent f42f6447d3
commit 988b87629e
1 changed files with 4 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
src/services/machineLearning/machineLearningService.ts
View file

@ -2,9 +2,9 @@ import { File, FILE_TYPE, getLocalFiles } from 'services/fileService';
import * as tf from '@tensorflow/tfjs-core'; import * as tf from '@tensorflow/tfjs-core';
import '@tensorflow/tfjs-backend-webgl'; import '@tensorflow/tfjs-backend-webgl';
import '@tensorflow/tfjs-backend-wasm'; // import '@tensorflow/tfjs-backend-wasm';
import { setWasmPaths } from '@tensorflow/tfjs-backend-wasm'; // import { setWasmPaths } from '@tensorflow/tfjs-backend-wasm';
import '@tensorflow/tfjs-backend-cpu'; // import '@tensorflow/tfjs-backend-cpu';
import { import {
DetectedFace, DetectedFace,
@ -51,7 +51,7 @@ class MachineLearningService {
private localSyncContext: Promise<MLSyncContext>; private localSyncContext: Promise<MLSyncContext>;
public constructor() { public constructor() {
setWasmPaths('/js/tfjs/'); // setWasmPaths('/js/tfjs/');
// this.faceDetectionService = new TFJSFaceDetectionService(); // this.faceDetectionService = new TFJSFaceDetectionService();
// this.faceLandmarkService = new FAPIFaceLandmarksService(); // this.faceLandmarkService = new FAPIFaceLandmarksService();
// this.faceAlignmentService = new ArcfaceAlignmentService(); // this.faceAlignmentService = new ArcfaceAlignmentService();