Mercurial Hosting > traffic-intelligence
comparison 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 |
comparison
equal
deleted
inserted
replaced
894:0c1fed9e8862 | 895:739acd338cc0 |
---|---|
1 #! /usr/bin/env python | |
2 | |
3 import argparse | |
4 | |
5 import storage, cvutils | |
6 | |
7 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/).') | |
8 parser.add_argument('-i', dest = 'filename', help = 'filename of the camera calibration (.tacal)', required = True) | |
9 parser.add_argument('-o', dest = 'outputIntrinsicFilename', help = 'filename of the intrinsic camera matrix', default = 'intrinsic-camera.txt') | |
10 | |
11 args = parser.parse_args() | |
12 | |
13 cameraData = storage.loadPinholeCameraModel(args.filename, True) | |
14 if cameraData is not None: | |
15 from numpy import savetxt | |
16 intrinsicCameraMatrix = cvutils.getIntrinsicCameraMatrix(cameraData) | |
17 distortionCoefficients = cvutils.getDistortionCoefficients(cameraData) | |
18 savetxt(args.outputIntrinsicFilename, intrinsicCameraMatrix) | |
19 print(distortionCoefficients) |