Mercurial Hosting > traffic-intelligence
diff scripts/extract-camera-parameters.py @ 895:739acd338cc0
added script to extract camera info from tacal file by Lund University (T analyst)
author | Nicolas Saunier <nicolas.saunier@polymtl.ca> |
---|---|
date | Thu, 01 Jun 2017 14:54:11 -0400 |
parents | |
children | 933670761a57 |
line wrap: on
line diff
--- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/scripts/extract-camera-parameters.py Thu Jun 01 14:54:11 2017 -0400 @@ -0,0 +1,19 @@ +#! /usr/bin/env python + +import argparse + +import storage, cvutils + +parser = argparse.ArgumentParser(description='The program extracts the intrinsic camera from the tacal camera calibration file used by T-Analyst (http://www.tft.lth.se/en/research/video-analysis/co-operation/software/t-analyst/).') +parser.add_argument('-i', dest = 'filename', help = 'filename of the camera calibration (.tacal)', required = True) +parser.add_argument('-o', dest = 'outputIntrinsicFilename', help = 'filename of the intrinsic camera matrix', default = 'intrinsic-camera.txt') + +args = parser.parse_args() + +cameraData = storage.loadPinholeCameraModel(args.filename, True) +if cameraData is not None: + from numpy import savetxt + intrinsicCameraMatrix = cvutils.getIntrinsicCameraMatrix(cameraData) + distortionCoefficients = cvutils.getDistortionCoefficients(cameraData) + savetxt(args.outputIntrinsicFilename, intrinsicCameraMatrix) + print(distortionCoefficients)