Mercurial Hosting > traffic-intelligence
diff python/storage.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 | ff92801e5c54 |
children | 1466a63dd1cf |
line wrap: on
line diff
--- a/python/storage.py Thu Jun 01 10:33:07 2017 -0400 +++ b/python/storage.py Thu Jun 01 14:54:11 2017 -0400 @@ -1209,6 +1209,22 @@ out.close() +def loadPinholeCameraModel(filename, tanalystFormat = True): + '''Loads the data from a file containing the camera parameters + (pinhole camera model, http://docs.opencv.org/2.4/modules/calib3d/doc/camera_calibration_and_3d_reconstruction.html) + and returns a dictionary''' + if tanalystFormat: + f = openCheck(filename, quitting = True) + content = getLines(f) + cameraData = {} + for l in content: + tmp = l.split(':') + cameraData[tmp[0]] = float(tmp[1].strip().replace(',','.')) + return cameraData + else: + print('Unknown camera model (not tanalyst format') + return None + def savePositionsToCsv(f, obj): timeInterval = obj.getTimeInterval() positions = obj.getPositions()