GstInference/Example pipelines with hierarchical metadata: Difference between revisions

GstInference/Example pipelines with hierarchical metadata (view source)

Revision as of 16:04, 17 March 2020

46 bytes added , 17 March 2020

no edit summary

Ctrejos

352

edits

@@ Line 306: / Line 306: @@
    tinyyolov3: "output_boxes",
    facenetv1: "output"
-};
-var model_names = {
-  inceptionv1: "graph_inceptionv1_tensorflow.pb",
-  inceptionv2: "graph_inceptionv2_tensorflow.pb",
-  inceptionv3: "graph_inceptionv3_tensorflow.pb",
-  inceptionv4: "graph_inceptionv4_tensorflow.pb",
-  mobilenetv2: "graph_mobilenetv2_tensorflow.pb",
-  resnet50v1: "graph_resnetv1_tensorflow.pb",
-  tinyyolov2: "graph_tinyyolov2_tensorflow.pb",
-  tinyyolov3: "graph_tinyyolov3_tensorflow.pb",
-  facenetv1: "graph_facenetv1_tensorflow.pb"
 };
@@ Line 424: / Line 412: @@
        disable_element("outputlayer");
        break;
+  }
+  if (document.getElementById("model").value != "") {
+    model_selection();
    }
 }
 function model_selection() {
-   if( document.getElementById("backend").value == "tensorflow") {
+   document.getElementById("labels").value = label_files[document.getElementById("model").value];
-     document.getElementById("inputlayer").value = input_layers[document.getElementById("model").value];
+  var tmp_model_location = "";
-    document.getElementById("outputlayer").value = output_layers[document.getElementById("model").value];
+  var tmp_backend = document.getElementById("backend").value;
+  switch (tmp_backend) {
+     case "tensorflow":
+      document.getElementById("inputlayer").value = input_layers[document.getElementById("model").value];
+      document.getElementById("outputlayer").value = output_layers[document.getElementById("model").value];
+      tmp_model_location = "graph_" +  document.getElementById("model").value + "_tensorflow.pb";
+    break;
+    case "tflite":
+      tmp_model_location = "graph_" +  document.getElementById("model").value + "_tflite.pb";
+    break;
+    case "ncsdk":
+      tmp_model_location = "graph_" +  document.getElementById("model").value + "_ncsdk";
+    break;
+    default:
+      tmp_model_location = "";
+    break;
    }
-   document.getElementById("model_location").value = model_names[document.getElementById("model").value];
+   document.getElementById("model_location").value = tmp_model_location;
-  document.getElementById("labels").value = label_files[document.getElementById("model").value];
 }
@@ Line 452: / Line 457: @@
 }
-// TODO: modify for platform
 function tee_selection() {
    if (model != "") {