opencv/opencv_contrib
1
0
Fork 0
mirror of https://github.com/opencv/opencv_contrib.git synced 2025-10-16 22:35:51 +08:00
Code Issues Releases Wiki Activity

Merge remote-tracking branch 'upstream/3.4' into merge-3.4

Browse Source
This commit is contained in:
Alexander Alekhin
2019-09-25 18:55:28 +03:00
parent 5faa3ec873 1a15cbc9ae
commit 26129cfe69
1 changed files with 6 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
modules/optflow/samples/motempl.py
View File

@@ -11,10 +11,11 @@ MIN_TIME_DELTA = 0.05
def nothing(dummy): def nothing(dummy):
pass pass
def draw_motion_comp(vis, (x, y, w, h), angle, color): def draw_motion_comp(vis, rect, angle, color):
x, y, w, h = rect
cv.rectangle(vis, (x, y), (x+w, y+h), (0, 255, 0)) cv.rectangle(vis, (x, y), (x+w, y+h), (0, 255, 0))
r = min(w/2, h/2) r = min(w//2, h//2)
cx, cy = x+w/2, y+h/2 cx, cy = x+w//2, y+h//2
angle = angle*np.pi/180 angle = angle*np.pi/180
cv.circle(vis, (cx, cy), r, color, 3) cv.circle(vis, (cx, cy), r, color, 3)
cv.line(vis, (cx, cy), (int(cx+np.cos(angle)*r), int(cy+np.sin(angle)*r)), color, 3) cv.line(vis, (cx, cy), (int(cx+np.cos(angle)*r), int(cy+np.sin(angle)*r)), color, 3)
@@ -91,4 +92,6 @@ if __name__ == '__main__':
prev_frame = frame.copy() prev_frame = frame.copy()
if 0xFF & cv.waitKey(5) == 27: if 0xFF & cv.waitKey(5) == 27:
break break
# cleanup the camera and close any open windows
cam.release()
cv.destroyAllWindows() cv.destroyAllWindows()