352
edits
mNo edit summary |
No edit summary |
||
Line 306: | Line 306: | ||
tinyyolov3: "output_boxes", | tinyyolov3: "output_boxes", | ||
facenetv1: "output" | facenetv1: "output" | ||
}; | }; | ||
Line 424: | Line 412: | ||
disable_element("outputlayer"); | disable_element("outputlayer"); | ||
break; | break; | ||
} | |||
if (document.getElementById("model").value != "") { | |||
model_selection(); | |||
} | } | ||
} | } | ||
function model_selection() { | function model_selection() { | ||
document.getElementById("labels").value = label_files[document.getElementById("model").value]; | |||
document.getElementById("inputlayer").value = input_layers[document.getElementById("model").value]; | var tmp_model_location = ""; | ||
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 = | document.getElementById("model_location").value = tmp_model_location; | ||
} | } | ||
Line 452: | Line 457: | ||
} | } | ||
function tee_selection() { | function tee_selection() { | ||
if (model != "") { | if (model != "") { |
edits