learning_model 1.0.32 → 1.0.33

package/dist/learning/mobilenet.js

@@ -291,13 +291,22 @@ class LearningMobilenet {
         return __awaiter(this, void 0, void 0, function* () {
             try {
                 const truncatedModel = yield this.loadModel();
-                console.log('truncatedModel', truncatedModel, truncatedModel.outputs[0].shape.slice(1));
                 // 입력 이미지 크기에 맞게 모델 구조 수정
                 const model = tf.sequential();
-                model.add(tf.layers.flatten({
+                // 추가적인 합성곱 층
+                model.add(tf.layers.conv2d({
+                    filters: 64,
+                    kernelSize: 3,
+                    activation: 'relu',
+                    padding: 'same',
                     inputShape: truncatedModel.outputs[0].shape.slice(1)
                 }));
+                model.add(tf.layers.flatten());
+                // 더 깊은 밀집층
                 model.add(tf.layers.dense({ units: 100, activation: 'relu' }));
+                model.add(tf.layers.dense({ units: 100, activation: 'relu' }));
+                // 드롭아웃 층
+                model.add(tf.layers.dropout({ rate: 0.5 }));
                 model.add(tf.layers.dense({ units: this.labels.length, activation: 'softmax' }));
                 const optimizer = tf.train.adam(this.learningRate); // Optimizer를 생성하고 학습률을 설정합니다.
                 model.compile({

package/lib/learning/mobilenet.ts

@@ -293,15 +293,29 @@ class LearningMobilenet implements LearningInterface {
   private async _createModel(numClasses: number): Promise<tf.Sequential> {
     try {
         const truncatedModel = await this.loadModel()
-        console.log('truncatedModel', truncatedModel, truncatedModel.outputs[0].shape.slice(1));
         // 입력 이미지 크기에 맞게 모델 구조 수정
         const model = tf.sequential();
-        model.add(tf.layers.flatten({
+        // 추가적인 합성곱 층
+        model.add(tf.layers.conv2d({
+          filters: 64,
+          kernelSize: 3,
+          activation: 'relu',
+          padding: 'same',
           inputShape: truncatedModel.outputs[0].shape.slice(1)
         }));
+
+        model.add(tf.layers.flatten());
+
+        // 더 깊은 밀집층
+        model.add(tf.layers.dense({ units: 100, activation: 'relu' }));
         model.add(tf.layers.dense({ units: 100, activation: 'relu' }));
+
+        // 드롭아웃 층
+        model.add(tf.layers.dropout({ rate: 0.5 }));
+
         model.add(tf.layers.dense({ units: this.labels.length, activation: 'softmax' }));
 
+
         const optimizer = tf.train.adam(this.learningRate); // Optimizer를 생성하고 학습률을 설정합니다.
         model.compile({
             loss: (numClasses === 2) ? 'binaryCrossentropy': 'categoricalCrossentropy',

package/package.json

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