|
|
@ -66,8 +66,8 @@ namespace ofxOnnxRuntime |
|
|
|
// 4. output names & output dimms
|
|
|
|
num_outputs = ort_session->GetOutputCount(); |
|
|
|
output_node_names.resize(num_outputs); |
|
|
|
output_node_dims.clear(); |
|
|
|
output_values.clear(); |
|
|
|
output_node_dims.clear(); |
|
|
|
output_values.clear(); |
|
|
|
for (unsigned int i = 0; i < num_outputs; ++i) |
|
|
|
{ |
|
|
|
output_node_names[i] = ort_session->GetOutputName(i, allocator); |
|
|
@ -75,7 +75,7 @@ namespace ofxOnnxRuntime |
|
|
|
auto output_tensor_info = output_type_info.GetTensorTypeAndShapeInfo(); |
|
|
|
auto output_dims = output_tensor_info.GetShape(); |
|
|
|
output_node_dims.emplace_back(output_dims); |
|
|
|
output_values.emplace_back(nullptr); |
|
|
|
output_values.emplace_back(nullptr); |
|
|
|
} |
|
|
|
} |
|
|
|
|
|
|
|