learning_model 1.0.50 → 1.0.51

package/dist/index.bundle.js.LICENSE.txt

@@ -105,6 +105,23 @@
  * =============================================================================
  */
 
+/**
+ * @license
+ * Copyright 2019 Google LLC. All Rights Reserved.
+ * Licensed under the Apache License, Version 2.0 (the 'License');
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an 'AS IS' BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ * =============================================================================
+ */
+
 /**
  * @license
  * Copyright 2020 Google Inc. All Rights Reserved.

package/dist/learning/mobilenet.d.ts

@@ -1,4 +1,5 @@
 import * as tf from '@tensorflow/tfjs';
+import '@tensorflow/tfjs-backend-wasm';
 import { io } from '@tensorflow/tfjs-core';
 import LearningInterface from './base';
 declare class LearningMobilenet implements LearningInterface {
@@ -39,6 +40,8 @@ declare class LearningMobilenet implements LearningInterface {
     private _convertToTfDataset;
     addData(label: string, data: any): Promise<void>;
     init(): Promise<void>;
+    private setupBackend;
+    private checkWasmSupport;
     train(): Promise<tf.History>;
     infer(data: any): Promise<Map<string, number>>;
     saveModel(handlerOrURL: io.IOHandler | string, config?: io.SaveConfig): Promise<void>;

package/dist/learning/mobilenet.js

@@ -38,6 +38,7 @@ var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, ge
 };
 Object.defineProperty(exports, "__esModule", { value: true });
 const tf = __importStar(require("@tensorflow/tfjs"));
+require("@tensorflow/tfjs-backend-wasm"); // WebAssembly 백엔드 추가
 const tfjs_1 = require("@tensorflow/tfjs");
 const tf_1 = require("../utils/tf");
 const canvas_1 = require("../utils/canvas");
@@ -172,6 +173,8 @@ class LearningMobilenet {
     init() {
         return __awaiter(this, void 0, void 0, function* () {
             try {
+                console.log('init call');
+                yield this.setupBackend();
                 this.mobilenetModule = yield (0, tf_1.loadModel)();
             }
             catch (error) {
@@ -180,6 +183,34 @@ class LearningMobilenet {
             }
         });
     }
+    setupBackend() {
+        return __awaiter(this, void 0, void 0, function* () {
+            const isWasmSupported = yield this.checkWasmSupport();
+            if (isWasmSupported) {
+                yield tf.setBackend('wasm');
+                yield tf.ready();
+                console.log('Backend is set to WebAssembly');
+            }
+            else {
+                yield tf.setBackend('cpu');
+                yield tf.ready();
+                console.log('Backend is set to CPU');
+            }
+        });
+    }
+    checkWasmSupport() {
+        return __awaiter(this, void 0, void 0, function* () {
+            try {
+                yield tf.setBackend('wasm');
+                yield tf.ready();
+                return true;
+            }
+            catch (error) {
+                console.warn('WASM backend is not supported in this environment.', error);
+                return false;
+            }
+        });
+    }
     // 모델 학습 처리
     train() {
         return __awaiter(this, void 0, void 0, function* () {

package/dist/lib/learning/mobilenet.d.ts

@@ -1,4 +1,5 @@
 import * as tf from '@tensorflow/tfjs';
+import '@tensorflow/tfjs-backend-wasm';
 import { io } from '@tensorflow/tfjs-core';
 import LearningInterface from './base';
 declare class LearningMobilenet implements LearningInterface {
@@ -39,6 +40,8 @@ declare class LearningMobilenet implements LearningInterface {
     private _convertToTfDataset;
     addData(label: string, data: any): Promise<void>;
     init(): Promise<void>;
+    private setupBackend;
+    private checkWasmSupport;
     train(): Promise<tf.History>;
     infer(data: any): Promise<Map<string, number>>;
     saveModel(handlerOrURL: io.IOHandler | string, config?: io.SaveConfig): Promise<void>;

package/lib/learning/mobilenet.ts

@@ -5,6 +5,7 @@
 ///////////////////////////////////////////////////////////////////////////
 
 import * as tf from '@tensorflow/tfjs';
+import '@tensorflow/tfjs-backend-wasm';  // WebAssembly 백엔드 추가
 import { dispose } from '@tensorflow/tfjs';
 import { io } from '@tensorflow/tfjs-core';
 import LearningInterface from './base';
@@ -188,6 +189,8 @@ class LearningMobilenet implements LearningInterface {
 
   public async init() {
     try {
+      console.log('init call')
+      await this.setupBackend();
       this.mobilenetModule = await loadModel();
     } catch(error) {
       console.log('init Error', error);
@@ -195,6 +198,29 @@ class LearningMobilenet implements LearningInterface {
     } 
   } 
 
+  private async setupBackend() {
+    const isWasmSupported = await this.checkWasmSupport();
+    if (isWasmSupported) {
+      await tf.setBackend('wasm');
+      await tf.ready();
+      console.log('Backend is set to WebAssembly');
+    } else {
+      await tf.setBackend('cpu');
+      await tf.ready();
+      console.log('Backend is set to CPU');
+    }
+  }
+
+  private async checkWasmSupport(): Promise<boolean> {
+    try {
+      await tf.setBackend('wasm');
+      await tf.ready();
+      return true;
+    } catch (error) {
+      console.warn('WASM backend is not supported in this environment.', error);
+      return false;
+    }
+  }
 
   // 모델 학습 처리
   public async train(): Promise<tf.History> {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "learning_model",
-  "version": "1.0.50",
+  "version": "1.0.51",
   "description": "learning model develop",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",
@@ -20,6 +20,7 @@
     "@tensorflow-models/mobilenet": "^2.1.0",
     "@tensorflow/tfjs": "^4.6.0",
     "@tensorflow/tfjs-layers": "^4.6.0",
+    "@tensorflow/tfjs-backend-wasm": "^4.20.0",
     "canvas": "^2.11.2",
     "learning_model": "^1.0.0"
   },