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)