Mercurial Hosting > traffic-intelligence

comparison scripts/compute-homography.py @ 697:0421a5a0072c dev

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
improvemend by Adrien Lessard to choose output homography filename
author Nicolas Saunier <nicolas.saunier@polymtl.ca>
date Fri, 17 Jul 2015 14:22:45 -0400
parents fe34c0f79c32
children bf4a1790cfac
comparison
equal deleted inserted replaced
696:ae137e3b1990 697:0421a5a0072c
24 parser.add_argument('--tsai', dest = 'tsaiCameraFilename', help = 'name of the text file containing the camera parameter following the pinhole camera model (Lund format)') # caution, this is Aliaksei's format 24 parser.add_argument('--tsai', dest = 'tsaiCameraFilename', help = 'name of the text file containing the camera parameter following the pinhole camera model (Lund format)') # caution, this is Aliaksei's format
25 parser.add_argument('-i', dest = 'videoFrameFilename', help = 'filename of the video frame') 25 parser.add_argument('-i', dest = 'videoFrameFilename', help = 'filename of the video frame')
26 parser.add_argument('-w', dest = 'worldFilename', help = 'filename of the aerial photo/ground map') 26 parser.add_argument('-w', dest = 'worldFilename', help = 'filename of the aerial photo/ground map')
27 parser.add_argument('-n', dest = 'nPoints', help = 'number of corresponding points to input', default = 4, type = int) 27 parser.add_argument('-n', dest = 'nPoints', help = 'number of corresponding points to input', default = 4, type = int)
28 parser.add_argument('-u', dest = 'unitsPerPixel', help = 'number of units per pixel', default = 1., type = float) 28 parser.add_argument('-u', dest = 'unitsPerPixel', help = 'number of units per pixel', default = 1., type = float)
29 parser.add_argument('-o', dest = 'homographyFilename', help = 'filename of the homography matrix', default = 'homography.txt')
29 parser.add_argument('--display', dest = 'displayPoints', help = 'display original and projected points on both images', action = 'store_true') 30 parser.add_argument('--display', dest = 'displayPoints', help = 'display original and projected points on both images', action = 'store_true')
30 parser.add_argument('--intrinsic', dest = 'intrinsicCameraMatrixFilename', help = 'name of the intrinsic camera file') 31 parser.add_argument('--intrinsic', dest = 'intrinsicCameraMatrixFilename', help = 'name of the intrinsic camera file')
31 parser.add_argument('--distortion-coefficients', dest = 'distortionCoefficients', help = 'distortion coefficients', nargs = '*', type = float) 32 parser.add_argument('--distortion-coefficients', dest = 'distortionCoefficients', help = 'distortion coefficients', nargs = '*', type = float)
32 parser.add_argument('--undistorted-multiplication', dest = 'undistortedImageMultiplication', help = 'undistorted image multiplication', type = float) 33 parser.add_argument('--undistorted-multiplication', dest = 'undistortedImageMultiplication', help = 'undistorted image multiplication', type = float)
33 parser.add_argument('--undistort', dest = 'undistort', help = 'undistort the video (because features have been extracted that way', action = 'store_true') 34 parser.add_argument('--undistort', dest = 'undistort', help = 'undistort the video (because features have been extracted that way', action = 'store_true')
111 np.savetxt(f, worldPts.T) 112 np.savetxt(f, worldPts.T)
112 np.savetxt(f, videoPts.T) 113 np.savetxt(f, videoPts.T)
113 f.close() 114 f.close()
114 115
115 if homography.size>0: 116 if homography.size>0:
116 np.savetxt('homography.txt',homography) 117 np.savetxt(args.homographyFilename,homography)
117 118
118 if args.displayPoints and args.videoFrameFilename is not None and args.worldFilename is not None and homography.size>0 and args.tsaiCameraFilename is None: 119 if args.displayPoints and args.videoFrameFilename is not None and args.worldFilename is not None and homography.size>0 and args.tsaiCameraFilename is None:
119 worldImg = cv2.imread(args.worldFilename) 120 worldImg = cv2.imread(args.worldFilename)
120 videoImg = cv2.imread(args.videoFrameFilename) 121 videoImg = cv2.imread(args.videoFrameFilename)
121 if args.undistort: 122 if args.undistort: