diff scripts/rescale-homography.py @ 614:5e09583275a4

Merged Nicolas/trafficintelligence into default
author Mohamed Gomaa <eng.m.gom3a@gmail.com>
date Fri, 05 Dec 2014 12:13:53 -0500
parents 8f8f4375e441
children 56cc8a1f7082
line wrap: on
line diff
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/scripts/rescale-homography.py	Fri Dec 05 12:13:53 2014 -0500
@@ -0,0 +1,32 @@
+#! /usr/bin/env python
+
+import sys
+
+import matplotlib.pyplot as plt
+import numpy as np
+import cv2
+
+import cvutils
+import utils
+
+if len(sys.argv) < 4:
+   print('Usage: {} homography_filename original_size new_size (size can be width or height)'.format(sys.argv[0]))
+   sys.exit()
+
+homography = np.loadtxt(sys.argv[1])
+
+imgPoints = np.array([[10,10],
+                      [10,20],
+                      [20,20],
+                      [20,10]])
+
+wldPoints = cvutils.projectArray(homography, imgPoints.T).T
+
+newSize = float(sys.argv[3])
+originalSize = float(sys.argv[2])
+imgPoints = imgPoints*newSize/originalSize
+
+newHomography, mask = cv2.findHomography(imgPoints, wldPoints)
+
+np.savetxt(sys.argv[1]+'.new', newHomography)
+