Mercurial Hosting > traffic-intelligence
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) +