932
edits
Line 418: | Line 418: | ||
case "onnxrt" : | case "onnxrt" : | ||
search_option("tinyyolov3","TinyYolov3","model","add"); | search_option("tinyyolov3","TinyYolov3","model","add"); | ||
if( document.getElementById("model").value != "") { | if( document.getElementById("model").value != "") { | ||
disable_element("inputlayer"); | disable_element("inputlayer"); | ||
Line 453: | Line 452: | ||
break; | break; | ||
case "onnxrt": | case "onnxrt": | ||
tmp_model_location = "graph_" + document.getElementById("model").value + ".onnx"; | tmp_model_location = "graph_" + document.getElementById("model").value + ".onnx"; | ||
break; | break; | ||
Line 478: | Line 475: | ||
model_props = " name=net model-location=" + document.getElementById("model_location").value + " backend=" + document.getElementById("backend").value + " labels=\"$(cat " +document.getElementById("labels").value + ")\""; | model_props = " name=net model-location=" + document.getElementById("model_location").value + " backend=" + document.getElementById("backend").value + " labels=\"$(cat " +document.getElementById("labels").value + ")\""; | ||
if | if (backend == "tensorflow") { | ||
model_props = model_props + " backend::input-layer=" + document.getElementById("inputlayer").value + " backend::output-layer=" + document.getElementById("outputlayer").value; | model_props = model_props + " backend::input-layer=" + document.getElementById("inputlayer").value + " backend::output-layer=" + document.getElementById("outputlayer").value; | ||
} | } |
edits