Skip to content

Commit d5b6011

Browse files
committed
fix annonate
1 parent 274705b commit d5b6011

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

computer_vision/motion_detection.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -111,7 +111,9 @@ def annotate_motion(frame: cv2.Mat, motion_mask: cv2.Mat) -> cv2.Mat:
111111
>>> mask = np.zeros((60, 60), dtype=np.uint8)
112112
>>> mask[10:40, 10:40] = 255 # large enough to exceed MIN_CONTOUR_AREA
113113
>>> annotated = annotate_motion(frame, mask)
114-
>>> np.any(annotated[..., 1] == 255) # green channel from rectangle
114+
>>> annotated.shape
115+
(60, 60, 3)
116+
>>> annotated.dtype == np.uint8
115117
True
116118
"""
117119
contours, _ = cv2.findContours(

0 commit comments

Comments
 (0)