932
edits
No edit summary |
|||
Line 453: | Line 453: | ||
document.getElementById("outputlayer").value = output_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"; | tmp_model_location = "graph_" + document.getElementById("model").value + "_tensorflow.pb"; | ||
break; | |||
case "onnxrt": | |||
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 + "_onnx.onnx"; | |||
break; | break; | ||
case "tflite": | case "tflite": |
edits