From: Antonio Ospite Date: Mon, 18 Dec 2017 12:33:15 +0000 (+0100) Subject: gst-test-perspective.py: port to the cv2 module X-Git-Url: https://git.ao2.it/experiments/gstreamer.git/commitdiff_plain/a1bef368cff17b0f69f0b138fac329f3769dbe83?ds=sidebyside gst-test-perspective.py: port to the cv2 module --- 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():