Skip to content

Commit bd383a4

Browse files
committed
Fix batch indexing
1 parent ff90bd8 commit bd383a4

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

demos/interactive_face_detection_demo/cpp/detectors.cpp

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -248,7 +248,7 @@ AgeGenderDetection::AgeGenderDetection(const std::string &pathToModel,
248248
void AgeGenderDetection::submitRequest() {
249249
if (!enquedFaces)
250250
return;
251-
request.set_input_tensor(ov::Tensor{request.get_input_tensor(), {0, 0, 0, 0}, {enquedFaces + 1, inShape[1], inShape[2], inShape[3]}});
251+
request.set_input_tensor(ov::Tensor{request.get_input_tensor(), {0, 0, 0, 0}, {enquedFaces, inShape[1], inShape[2], inShape[3]}});
252252
request.start_async();
253253
enquedFaces = 0;
254254
}
@@ -306,7 +306,7 @@ HeadPoseDetection::HeadPoseDetection(const std::string &pathToModel,
306306

307307
void HeadPoseDetection::submitRequest() {
308308
if (!enquedFaces) return;
309-
request.set_input_tensor(ov::Tensor{request.get_input_tensor(), {0, 0, 0, 0}, {enquedFaces + 1, inShape[1], inShape[2], inShape[3]}});
309+
request.set_input_tensor(ov::Tensor{request.get_input_tensor(), {0, 0, 0, 0}, {enquedFaces, inShape[1], inShape[2], inShape[3]}});
310310
request.start_async();
311311
enquedFaces = 0;
312312
}
@@ -363,7 +363,7 @@ EmotionsDetection::EmotionsDetection(const std::string &pathToModel,
363363

364364
void EmotionsDetection::submitRequest() {
365365
if (!enquedFaces) return;
366-
request.set_input_tensor(ov::Tensor{request.get_input_tensor(), {0, 0, 0, 0}, {enquedFaces + 1, inShape[1], inShape[2], inShape[3]}});
366+
request.set_input_tensor(ov::Tensor{request.get_input_tensor(), {0, 0, 0, 0}, {enquedFaces, inShape[1], inShape[2], inShape[3]}});
367367
request.start_async();
368368
enquedFaces = 0;
369369
}

0 commit comments

Comments
 (0)