learning_model 1.0.40 → 1.0.42

package/dist/learning/mobilenet.js

@@ -138,8 +138,7 @@ class LearningMobilenet {
         return __awaiter(this, void 0, void 0, function* () {
             try {
                 if (this.mobilenetModule !== null) {
-                    const croppedImage = (0, canvas_1.cropTo)(data, 224, false);
-                    const cap = (0, tf_1.isTensor)(data) ? data : (0, tf_1.capture)(croppedImage, false);
+                    const cap = (0, tf_1.isTensor)(data) ? data : (0, tf_1.capture)(data, false);
                     const predict = this.mobilenetModule.predict(cap);
                     const activation = predict.dataSync();
                     const classIndex = this.registerClassNumber(label);
@@ -278,10 +277,16 @@ class LearningMobilenet {
     saveModel(handlerOrURL, config) {
         var _a;
         return __awaiter(this, void 0, void 0, function* () {
-            if (!this.isTrainedDone) {
-                return Promise.reject(new Error('Train is not done status'));
+            try {
+                console.log('saveModel try', this.isTrainedDone);
+                if (!this.isTrainedDone) {
+                    return Promise.reject(new Error('Train is not done status'));
+                }
+                yield ((_a = this.model) === null || _a === void 0 ? void 0 : _a.save(handlerOrURL, config));
+            }
+            catch (e) {
+                console.log('saveModel Error', e);
             }
-            yield ((_a = this.model) === null || _a === void 0 ? void 0 : _a.save(handlerOrURL, config));
         });
     }
     // 진행중 여부

package/lib/learning/mobilenet.ts

@@ -150,8 +150,7 @@ class LearningMobilenet implements LearningInterface {
   public async addData(label: string, data: any): Promise<void> {
     try {
       if (this.mobilenetModule !== null) {
-        const croppedImage = cropTo(data, 224, false);
-        const cap = isTensor(data) ? data : capture(croppedImage, false);
+        const cap = isTensor(data) ? data : capture(data, false);
         const predict = this.mobilenetModule.predict(cap) as tf.Tensor;
         const activation = predict.dataSync() as Float32Array;
         const classIndex = this.registerClassNumber(label);
@@ -285,10 +284,15 @@ class LearningMobilenet implements LearningInterface {
   
   // 모델 저장
   public async saveModel(handlerOrURL: io.IOHandler | string, config?: io.SaveConfig): Promise<void> {
-    if (!this.isTrainedDone) {
-      return Promise.reject(new Error('Train is not done status'));
+    try {
+      console.log('saveModel try', this.isTrainedDone);
+      if (!this.isTrainedDone) {
+        return Promise.reject(new Error('Train is not done status'));
+      }
+      await this.model?.save(handlerOrURL, config);
+    } catch(e) {
+      console.log('saveModel Error', e);
     }
-    await this.model?.save(handlerOrURL, config);
   }
 
   // 진행중 여부

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "learning_model",
-  "version": "1.0.40",
+  "version": "1.0.42",
   "description": "learning model develop",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",