ONNX-Runtime-Inference icon indicating copy to clipboard operation
ONNX-Runtime-Inference copied to clipboard

multiple input

Open Blaise-wig opened this issue 4 years ago • 1 comments

Hello, my question is, what if your input is not an image but for example more than one sensor data. how do perform inference wen you already have your onnx file? thank you

Blaise-wig avatar Feb 07 '22 13:02 Blaise-wig

I think this may help you:

` std::vectorOrt::Value inputTensors; std::vector<std::vector> inputTensorsValues; std::vector inputDims; int numInputs = session.GetInputCount(); std::vector<const char*> inputNames;

	for (int i = 0; i < numInputs; ++i) {
		inputNames.push_back(session.GetInputName(i, allocator));

		std::vector<int64_t> inputDimsI = session.GetInputTypeInfo(i).GetTensorTypeAndShapeInfo().GetShape();
		int inputTensorSize = vectorProduct(inputDimsI);
		inputDims.push_back(inputTensorSize);
		std::vector<float> inputTensorValues(inputTensorSize);
		inputTensorsValues.push_back(inputTensorValues);
		inputTensors.push_back(Ort::Value::CreateTensor<float>(
			memoryInfo, inputTensorsValues[i].data(), inputTensorSize,
			inputDimsI.data(), inputDimsI.size()));
	}`

EnriqueFdez03 avatar Mar 07 '22 12:03 EnriqueFdez03