ort.run( session as Node, inputs as Node, [options as Object] ) as Object
Perform inference of a session, based on supplied input values. Returns an
Object
of output names and their values.
const session = ort.session(cts.doc("testmodel.onnx")) const inputCount = ort.sessionInputCount(session) const outputCount = ort.sessionOutputCount(session) var inputNames = [] var i,j for (i=0;i<inputCount;i++){ inputNames.push(ort.sessionInputName(session, i)) } var outputNames = [] for (i=0;i<outputCount;i++){ outputNames.push(ort.sessionOutputName(session, i)) } var inputTypes = [] for (i=0;i<inputCount;i++){ inputTypes.push(ort.sessionInputType(session, i)) } var outputTypes = [] for (i=0;i<outputCount;i++){ outputTypes.push(ort.sessionOutputType(session, i)) } var inputValues = [] for (i=0;i<inputCount;i++){ var p = 1 for(j=0;j<inputTypes[i]["shape"].length;j++){ p *= inputTypes[i]["shape"][j] } var data = [] for(j=0;j<p;j++){ data.push(j); } inputValues.push(ort.value(data, inputTypes[i]["shape"], "float")) } var inputMap = {} for (i=0;i<inputCount;i++){ inputMap[inputNames[i]] = inputValues[i] } ort.run(session, inputMap) => { "output_0": "OrtValue(Shape:[1, 1000, 1, 1], Type: FLOAT)" }
Stack Overflow: Get the most useful answers to questions from the MarkLogic community, or ask your own question.