Update index.html

index.html

@@ -338,17 +338,17 @@
         </div>
     </div>
 </div>
-
 <script>
-    // --- 変更部分: モデルリストに仕様(type)を明記して定義 ---
+    // モデル定義（新モデル tinymodelV9s.onnx を追加）
     const models = [
+        { file: "tinymodelV9s.onnx", type: "auto" },    // 自動判定
         { file: "tinymodelV3.onnx", type: "hsc" },     // 異常検知スコア (1出力)
         { file: "tinymodelV2.onnx", type: "softmax" } // 通常の分類 (2出力)
     ];
 
     let session = null;
     let isReady = false;
-    let currentModelConfig = null; // 現在選択されているモデルの設定を保持
+    let currentModelConfig = null;
 
     const modelSelect = document.getElementById('modelSelect');
     const fileInput = document.getElementById('fileInput');
@@ -360,13 +360,13 @@
     const resultOutput = document.getElementById('resultOutput');
 
     function initSelect() {
+        modelSelect.innerHTML = '';
         models.forEach((m, index) => {
             const opt = document.createElement('option');
-            opt.value = index; // インデックスをvalueに設定
+            opt.value = index;
             opt.textContent = m.file;
             modelSelect.appendChild(opt);
         });
-        // 選択時に該当するオブジェクトを渡す
         modelSelect.addEventListener('change', (e) => loadModel(models[e.target.value]));
     }
 
@@ -447,7 +447,7 @@
             const output = results[session.outputNames[0]].data;
             let probNeed, probTrash;
 
-            // --- 事前定義されたモデルの仕様(type)に基づいて計算を分岐 ---
+            // モデル仕様に応じた確率計算の分岐
             if (currentModelConfig.type === "hsc") {
                 const trashScore = output[0];
                 probTrash = trashScore * 100;
@@ -458,9 +458,25 @@
                 const sum = exp0 + exp1;
                 probNeed = (exp0 / sum) * 100;
                 probTrash = (exp1 / sum) * 100;
+            } else {
+                // 自動判定 (出力数に応じて判定)
+                if (output.length === 1) {
+                    let score = output[0];
+                    if (score < 0 || score > 1) { // ロジットの場合は Sigmoid 変換
+                        score = 1 / (1 + Math.exp(-score));
+                    }
+                    probTrash = score * 100;
+                    probNeed = (1.0 - score) * 100;
+                } else {
+                    const exp0 = Math.exp(output[0]);
+                    const exp1 = Math.exp(output[1]);
+                    const sum = exp0 + exp1;
+                    probNeed = (exp0 / sum) * 100;
+                    probTrash = (exp1 / sum) * 100;
+                }
             }
 
-            // UI Render
+            // UI描画
             emptyOutput.style.display = 'none';
             resultOutput.style.display = 'flex';
             const topClass = probNeed >= probTrash ? "need" : "trash";