X-Git-Url: https://git.ao2.it/experiments/gstreamer.git/blobdiff_plain/1d15b020fc59912e53733be0c5cff1ef6ecbf6ef..a1bef368cff17b0f69f0b138fac329f3769dbe83:/python/gst-test-perspective.py?ds=sidebyside diff --git a/python/gst-test-perspective.py b/python/gst-test-perspective.py index b51591a..9a74823 100755 --- a/python/gst-test-perspective.py +++ b/python/gst-test-perspective.py @@ -2,7 +2,8 @@ # test program for the "perspective" geometric tranform element -import cv +import cv2 +import numpy as np import gi gi.require_version('Gst', '1.0') @@ -17,17 +18,14 @@ def calc_matrix(): top_x_shift = 200 width, height = 800, 480 - corners = [(0, 0), (width, 0), (width, height), (0, height)] - target = [(top_x_shift, 0), (width - top_x_shift, 0), (width, height), (0, height)] + corners = np.array([(0, 0), (width, 0), (width, height), (0, height)], dtype=np.float32) + target = np.array([(top_x_shift, 0), (width - top_x_shift, 0), (width, height), (0, height)], dtype=np.float32) - mat = cv.CreateMat(3, 3, cv.CV_64F) - cv.GetPerspectiveTransform(corners, target, mat) + mat = cv2.getPerspectiveTransform(corners, target) + ret, inv_mat = cv2.invert(mat) - inv_mat = cv.CreateMat(3, 3, cv.CV_64F) - cv.Invert(mat, inv_mat) + return inv_mat.flatten() - matrix = [inv_mat[j, i] for j in range(mat.rows) for i in range(mat.cols)] - return matrix def main():