File tree Expand file tree Collapse file tree 2 files changed +3
-3
lines changed Expand file tree Collapse file tree 2 files changed +3
-3
lines changed Original file line number Diff line number Diff line change @@ -47,6 +47,6 @@ class ModelYoloV3ONNX: public DetectionModel {
47
47
48
48
std::string boxesOutputName;
49
49
std::string scoresOutputName;
50
- std::string indicesOuputName ;
50
+ std::string indicesOutputName ;
51
51
static const int numberOfClasses = 80 ;
52
52
};
Original file line number Diff line number Diff line change @@ -97,7 +97,7 @@ void ModelYoloV3ONNX::prepareInputsOutputs(std::shared_ptr<ov::Model>& model) {
97
97
const ov::Shape& currentShape = output.get_partial_shape ().get_max_shape ();
98
98
std::string currentName = output.get_any_name ();
99
99
if (currentShape[currentShape.size () - 1 ] == 3 ) {
100
- indicesOuputName = currentName;
100
+ indicesOutputName = currentName;
101
101
ppp.output (currentName).tensor ().set_element_type (ov::element::i32 );
102
102
} else if (currentShape[2 ] == 4 ) {
103
103
boxesOutputName = currentName;
@@ -146,7 +146,7 @@ std::unique_ptr<ResultBase> ModelYoloV3ONNX::postprocess(InferenceResult& infRes
146
146
const float * boxesPtr = boxes.data <float >();
147
147
148
148
const ov::Tensor& scores = infResult.outputsData [scoresOutputName];
149
- const ov::Tensor& indices = infResult.outputsData [indicesOuputName ];
149
+ const ov::Tensor& indices = infResult.outputsData [indicesOutputName ];
150
150
151
151
const int * indicesData = indices.data <int >();
152
152
const auto indicesShape = indices.get_shape ();
You can’t perform that action at this time.
0 commit comments