update squeezenet demo

tafsiri · web-flow · commit bc95397cd8f8 · 2018-02-15T15:35:40.000-05:00
diff --git a/demos/imagenet/imagenet.html b/demos/imagenet/imagenet.html
@@ -89,8 +89,8 @@
 
     </div>
     <p>
-      This demo uses <a href="https://github.com/DeepScale/SqueezeNet"> Squeezenet 1.1 Model</a>
-      and is not trained to recognize human faces. For best performance, point your webcam at objects like piano, coffee mugs, bottles, etc.
+      This demo uses <a href="https://github.com/DeepScale/SqueezeNet"> Squeezenet 1.1 Model</a>. Explore what the activations
+      of the various squeezenet layers are using the dropdown below.
     </p>
   </div>
   <div id="ssl-message" style="display: none">
diff --git a/demos/imagenet/imagenet.ts b/demos/imagenet/imagenet.ts
@@ -204,13 +204,18 @@ export class ImagenetDemo extends ImagenetDemoPolymer {
 
       let count = 0;
       for (const className in topClassesToProbability) {
-        if (!(className in topClassesToProbability)) {
-          continue;
+        if (isWebcam) {
+          document.getElementById(`class${count}`).innerHTML = '';
+          document.getElementById(`prob${count}`).innerHTML = '';
+        } else {
+          if (!(className in topClassesToProbability)) {
+            continue;
+          }
+          document.getElementById(`class${count}`).innerHTML = className;
+          document.getElementById(`prob${count}`).innerHTML =
+              (Math.floor(1000 * topClassesToProbability[className]) / 1000)
+                  .toString();
         }
-        document.getElementById(`class${count}`).innerHTML = className;
-        document.getElementById(`prob${count}`).innerHTML =
-            (Math.floor(1000 * topClassesToProbability[className]) / 1000)
-                .toString();
         count++;
       }
 
diff --git a/demos/imagenet/index.html b/demos/imagenet/index.html
@@ -46,7 +46,7 @@
 </head>
 
 <body>
-  <demo-header name="webcam classifier"></demo-header>
+  <demo-header name="webcam squeezenet layer activations"></demo-header>
   <imagenet-demo></imagenet-demo>
   <demo-footer></demo-footer>
 </body>
