Add npu option to image classification

Honry · Oct 12, 2023 · 653d27d · 653d27d
1 parent 45f9a75
commit 653d27d
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/image_classification/index.html b/image_classification/index.html
@@ -40,6 +40,9 @@
               <label class="btn btn-outline-info custom" name="webnn">
                 <input type="radio" name="backend" id="webnn_cpu" autocomplete="off">WebNN (CPU)
               </label>
+              <label class="btn btn-outline-info custom" name="webnn">
+                <input type="radio" name="backend" id="webnn_npu" autocomplete="off">WebNN (NPU)
+              </label>
               <label class="btn btn-outline-info custom" name="webnn">
                 <input type="radio" name="backend" id="webnn_gpu" autocomplete="off">WebNN (GPU)
               </label>

diff --git a/image_classification/main.js b/image_classification/main.js
@@ -51,7 +51,7 @@ $(document).ready(async () => {
 
 $('#backendBtns .btn').on('change', async (e) => {
   if (inputType === 'camera') utils.stopCameraStream(rafReq, stream);
-  if ($(e.target).attr('id').indexOf('cpu') != -1) {
+  if ($(e.target).attr('id').indexOf('cpu') != -1 || $(e.target).attr('id').indexOf('npu') != -1) {
     layout = 'nhwc';
   } else if (($(e.target).attr('id').indexOf('gpu') != -1)) {
     layout = 'nchw';