Mercurial > hg > nsaunier > 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) |