@@ -225,22 +225,18 @@ std::vector<ModelFaceBoxes::Anchor> filterBoxes(const ov::Tensor& boxesTensor,
225
225
}
226
226
227
227
std::unique_ptr<ResultBase> ModelFaceBoxes::postprocess (InferenceResult& infResult) {
228
- // --------------------------- Filter scores and get valid indices for bounding
229
- // boxes----------------------------------
228
+ // Filter scores and get valid indices for bounding boxes
230
229
const auto scoresTensor = infResult.outputsData [outputsNames[1 ]];
231
230
const auto scores = filterScores (scoresTensor, confidenceThreshold);
232
231
233
- // --------------------------- Filter bounding boxes on indices
234
- // -------------------------------------------------------
232
+ // Filter bounding boxes on indices
235
233
auto boxesTensor = infResult.outputsData [outputsNames[0 ]];
236
234
std::vector<Anchor> boxes = filterBoxes (boxesTensor, anchors, scores.first , variance);
237
235
238
- // --------------------------- Apply Non-maximum Suppression
239
- // ----------------------------------------------------------
236
+ // Apply Non-maximum Suppression
240
237
const std::vector<int > keep = nms (boxes, scores.second , boxIOUThreshold);
241
238
242
- // --------------------------- Create detection result objects
243
- // --------------------------------------------------------
239
+ // Create detection result objects
244
240
DetectionResult* result = new DetectionResult (infResult.frameId , infResult.metaData );
245
241
const auto imgWidth = infResult.internalModelData ->asRef <InternalImageModelData>().inputImgWidth ;
246
242
const auto imgHeight = infResult.internalModelData ->asRef <InternalImageModelData>().inputImgHeight ;
0 commit comments