Update demos/common/cpp/models/src/detection_model_yolov3_onnx.cpp

akorobeinikov · Wovchena · web-flow · commit aa726e0fda0e · 2022-11-11T14:13:10.000+03:00
Co-authored-by: Zlobin Vladimir &lt;vladimir.zlobin@intel.com&gt;
diff --git a/demos/common/cpp/models/src/detection_model_yolov3_onnx.cpp b/demos/common/cpp/models/src/detection_model_yolov3_onnx.cpp
@@ -119,13 +119,15 @@ std::shared_ptr<InternalModelData> ModelYoloV3ONNX::preprocess(const InputData&
     return ImageModel::preprocess(inputData, request);
 }
 
+namespace {
 float getScore(const ov::Tensor& scoresTensor, size_t classInd, size_t boxInd) {
     const float* scoresPtr = scoresTensor.data<float>();
     const auto shape = scoresTensor.get_shape();
     int N = shape[2];
 
     return scoresPtr[classInd * N + boxInd];
 }
+}
 
 std::unique_ptr<ResultBase> ModelYoloV3ONNX::postprocess(InferenceResult& infResult) {
     // Get info about input image

Original file line number	Diff line number	Diff line change
`@@ -119,13 +119,15 @@ std::shared_ptr<InternalModelData> ModelYoloV3ONNX::preprocess(const InputData&`
`119`	`119`	`return ImageModel::preprocess(inputData, request);`
`120`	`120`	`}`
`121`	`121`
	`122`	`+namespace {`
`122`	`123`	`float getScore(const ov::Tensor& scoresTensor, size_t classInd, size_t boxInd) {`
`123`	`124`	`const float* scoresPtr = scoresTensor.data<float>();`
`124`	`125`	`const auto shape = scoresTensor.get_shape();`
`125`	`126`	`int N = shape[2];`
`126`	`127`
`127`	`128`	`return scoresPtr[classInd * N + boxInd];`
`128`	`129`	`}`
	`130`	`+}`
`129`	`131`
`130`	`132`	`std::unique_ptr<ResultBase> ModelYoloV3ONNX::postprocess(InferenceResult& infResult) {`
`131`	`133`	`// Get info about input image`