Commit 4f524db4 authored by Moeed's avatar Moeed Committed by Alexander Alekhin
Browse files

Merge pull request #2269 from shaikmoeed:hotfix/samples-motempl-python-update 

* Fix bug: Invalid Syntax at line 14(draw_motion_comp method)

* Fix bug: TypeError- integer argument expected, got float

* Fix bug: clean up the camera by adding .release()
parent f5e493b6

modules/optflow/samples/motempl.py

+6 −3
Original line number Diff line number Diff line
@@ -11,10 +11,11 @@ MIN_TIME_DELTA = 0.05 
def nothing(dummy):

    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))

    r = min(w/2, h/2)

    cx, cy = x+w/2, y+h/2

    r = min(w//2, h//2)

    cx, cy = x+w//2, y+h//2

    angle = angle*np.pi/180

    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)
@@ -91,4 +92,6 @@ if __name__ == '__main__': 
        prev_frame = frame.copy()

        if 0xFF & cv.waitKey(5) == 27:

            break

    # cleanup the camera and close any open windows

    cam.release()

    cv.destroyAllWindows()