File tree Expand file tree Collapse file tree 3 files changed +9
-6
lines changed
Linux/squeezenet_classification
Windows/squeezenet_classification Expand file tree Collapse file tree 3 files changed +9
-6
lines changed Original file line number Diff line number Diff line change 11/*
22Copyright (C) 2021, Intel Corporation
33SPDX-License-Identifier: Apache-2.0
4-
54Portions of this software are copyright of their respective authors and released under the MIT license:
65- ONNX-Runtime-Inference, Copyright 2020 Lei Mao. For licensing see https://github.com/leimao/ONNX-Runtime-Inference/blob/main/LICENSE.md
76*/
@@ -313,7 +312,7 @@ int main(int argc, char* argv[])
313312 so we have to prepare an array of Ort::Value instances for inputs and outputs respectively even if
314313 we only have one input and one output. */
315314
316- std::vector<const char *> inputNames{" data_0 " };
315+ std::vector<const char *> inputNames{inputName };
317316 std::vector<const char *> outputNames{outputName};
318317 std::vector<Ort::Value> inputTensors;
319318 std::vector<Ort::Value> outputTensors;
Original file line number Diff line number Diff line change @@ -260,7 +260,8 @@ int main(int argc, char* argv[])
260260 std::cout << " Number of Input Nodes: " << numInputNodes << std::endl;
261261 std::cout << " Number of Output Nodes: " << numOutputNodes << std::endl;
262262
263- const char * inputName = session.GetInputName (0 , allocator);
263+ auto inputNodeName = session.GetInputNameAllocated (0 , allocator);
264+ const char * inputName = inputNodeName.get ();
264265 std::cout << " Input Name: " << inputName << std::endl;
265266
266267 Ort::TypeInfo inputTypeInfo = session.GetInputTypeInfo (0 );
@@ -272,7 +273,8 @@ int main(int argc, char* argv[])
272273 std::vector<int64_t > inputDims = inputTensorInfo.GetShape ();
273274 std::cout << " Input Dimensions: " << inputDims << std::endl;
274275
275- const char * outputName = session.GetOutputName (0 , allocator);
276+ auto outputNodeName = session.GetOutputNameAllocated (0 , allocator);
277+ const char * outputName = outputNodeName.get ();
276278 std::cout << " Output Name: " << outputName << std::endl;
277279
278280 Ort::TypeInfo outputTypeInfo = session.GetOutputTypeInfo (0 );
Original file line number Diff line number Diff line change @@ -235,7 +235,8 @@ int main(int argc, char* argv[])
235235 std::cout << " Number of Input Nodes: " << numInputNodes << std::endl;
236236 std::cout << " Number of Output Nodes: " << numOutputNodes << std::endl;
237237
238- const char * inputName = session.GetInputName (0 , allocator);
238+ auto inputNodeName = session.GetInputNameAllocated (0 , allocator);
239+ const char * inputName = inputNodeName.get ();
239240 std::cout << " Input Name: " << inputName << std::endl;
240241
241242 Ort::TypeInfo inputTypeInfo = session.GetInputTypeInfo (0 );
@@ -247,7 +248,8 @@ int main(int argc, char* argv[])
247248 std::vector<int64_t > inputDims = inputTensorInfo.GetShape ();
248249 std::cout << " Input Dimensions: " << inputDims << std::endl;
249250
250- const char * outputName = session.GetOutputName (0 , allocator);
251+ auto outputNodeName = session.GetOutputNameAllocated (0 , allocator);
252+ const char * outputName = outputNodeName.get ();
251253 std::cout << " Output Name: " << outputName << std::endl;
252254
253255 Ort::TypeInfo outputTypeInfo = session.GetOutputTypeInfo (0 );
You can’t perform that action at this time.
0 commit comments