Mercurial Hosting > traffic-intelligence
view scripts/rescale-homography.py @ 1251:2b1c8fe8f7e4
added function for KITTI 2D format (just a simple csv-like format with xy coordinates)
author | Nicolas Saunier <nicolas.saunier@polymtl.ca> |
---|---|
date | Fri, 15 Mar 2024 17:05:54 -0400 |
parents | cc5cb04b04b0 |
children |
line wrap: on
line source
#! /usr/bin/env python3 import sys import matplotlib.pyplot as plt import numpy as np import cv2 from trafficintelligence import cvutils, 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.homographyProject(imgPoints.T, homography).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)