code/test_on_video_dlib.py
@@ -125,8 +125,8 @@ if conf > 1.0: bbox_width = abs(x_max - x_min) bbox_height = abs(y_max - y_min) x_min -= 3 * bbox_width / 4 x_max += 3 * bbox_width / 4 x_min -= 2 * bbox_width / 4 x_max += 2 * bbox_width / 4 y_min -= 3 * bbox_height / 4 y_max += bbox_height / 4 x_min = max(x_min, 0); y_min = max(y_min, 0) @@ -158,8 +158,8 @@ # Plot expanded bounding box cv2.rectangle(frame, (x_min, y_min), (x_max, y_max), (0,255,0), 1) out.write(frame) frame_num += 1 out.write(frame) frame_num += 1 out.release() video.release()
