Merge pull request #15516 from dkurt:dnn_detection_model_fix_3.4

pull/15540/head
Alexander Alekhin 5 years ago
commit 8167d47efc
  1. 2
      samples/dnn/object_detection.cpp
  2. 2
      samples/dnn/object_detection.py

@ -347,7 +347,7 @@ void postprocess(Mat& frame, const std::vector<Mat>& outs, Net& net)
int bottom = (int)data[i + 6];
int width = right - left + 1;
int height = bottom - top + 1;
if (width * height <= 1)
if (width <= 2 || height <= 2)
{
left = (int)(data[i + 3] * frame.cols);
top = (int)(data[i + 4] * frame.rows);

@ -126,7 +126,7 @@ def postprocess(frame, outs):
bottom = int(detection[6])
width = right - left + 1
height = bottom - top + 1
if width * height <= 1:
if width <= 2 or height <= 2:
left = int(detection[3] * frameWidth)
top = int(detection[4] * frameHeight)
right = int(detection[5] * frameWidth)

Loading…
Cancel
Save