Mercurial Hosting > traffic-intelligence
annotate scripts/compute-homography.py @ 750:6049e9b6902c dev
work in progress storage vissim sqlite
author | Nicolas Saunier <nicolas.saunier@polymtl.ca> |
---|---|
date | Wed, 28 Oct 2015 17:50:32 -0400 |
parents | 0421a5a0072c |
children | bf4a1790cfac |
rev | line source |
---|---|
151
4af774bb186d
wrote a simple script to compute homography from point correspondences and display the reprojection for visual verification
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
diff
changeset
|
1 #! /usr/bin/env python |
4af774bb186d
wrote a simple script to compute homography from point correspondences and display the reprojection for visual verification
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
diff
changeset
|
2 |
476
6551a3cf1750
modified compute-homography to work with argparse
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
443
diff
changeset
|
3 import sys, argparse |
151
4af774bb186d
wrote a simple script to compute homography from point correspondences and display the reprojection for visual verification
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
diff
changeset
|
4 |
238
be3761a09b20
added functions to input point correspondences
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
237
diff
changeset
|
5 import matplotlib.pyplot as plt |
151
4af774bb186d
wrote a simple script to compute homography from point correspondences and display the reprojection for visual verification
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
diff
changeset
|
6 import numpy as np |
4af774bb186d
wrote a simple script to compute homography from point correspondences and display the reprojection for visual verification
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
diff
changeset
|
7 import cv2 |
4af774bb186d
wrote a simple script to compute homography from point correspondences and display the reprojection for visual verification
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
diff
changeset
|
8 |
638
852f5de42d01
added functionality to read Aliaksei Tsai camera model data
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
636
diff
changeset
|
9 import cvutils, utils, storage |
852f5de42d01
added functionality to read Aliaksei Tsai camera model data
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
636
diff
changeset
|
10 |
852f5de42d01
added functionality to read Aliaksei Tsai camera model data
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
636
diff
changeset
|
11 # TODO add option to use RANSAC or other robust homography estimation method? |
151
4af774bb186d
wrote a simple script to compute homography from point correspondences and display the reprojection for visual verification
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
diff
changeset
|
12 |
476
6551a3cf1750
modified compute-homography to work with argparse
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
443
diff
changeset
|
13 parser = argparse.ArgumentParser(description='The program computes the homography matrix from at least 4 non-colinear point correspondences inputed in the same order in a video frame and a aerial photo/ground map, or from the list of corresponding points in the two planes.', epilog = '''The point correspondence file contains at least 4 non-colinear point coordinates |
6551a3cf1750
modified compute-homography to work with argparse
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
443
diff
changeset
|
14 with the following format: |
6551a3cf1750
modified compute-homography to work with argparse
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
443
diff
changeset
|
15 - the first two lines are the x and y coordinates in the projected space (usually world space) |
6551a3cf1750
modified compute-homography to work with argparse
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
443
diff
changeset
|
16 - the last two lines are the x and y coordinates in the origin space (usually image space) |
6551a3cf1750
modified compute-homography to work with argparse
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
443
diff
changeset
|
17 |
6551a3cf1750
modified compute-homography to work with argparse
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
443
diff
changeset
|
18 If providing video and world images, with a number of points to input |
6551a3cf1750
modified compute-homography to work with argparse
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
443
diff
changeset
|
19 and a ration to convert pixels to world distance unit (eg meters per pixel), |
6551a3cf1750
modified compute-homography to work with argparse
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
443
diff
changeset
|
20 the images will be shown in turn and the user should click |
591
aded6c1c2ebd
added framework script and function to compute matchings
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
572
diff
changeset
|
21 in the same order the corresponding points in world and image spaces.''', formatter_class=argparse.RawDescriptionHelpFormatter) |
476
6551a3cf1750
modified compute-homography to work with argparse
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
443
diff
changeset
|
22 |
6551a3cf1750
modified compute-homography to work with argparse
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
443
diff
changeset
|
23 parser.add_argument('-p', dest = 'pointCorrespondencesFilename', help = 'name of the text file containing the point correspondences') |
638
852f5de42d01
added functionality to read Aliaksei Tsai camera model data
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
636
diff
changeset
|
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 |
476
6551a3cf1750
modified compute-homography to work with argparse
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
443
diff
changeset
|
25 parser.add_argument('-i', dest = 'videoFrameFilename', help = 'filename of the video frame') |
6551a3cf1750
modified compute-homography to work with argparse
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
443
diff
changeset
|
26 parser.add_argument('-w', dest = 'worldFilename', help = 'filename of the aerial photo/ground map') |
6551a3cf1750
modified compute-homography to work with argparse
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
443
diff
changeset
|
27 parser.add_argument('-n', dest = 'nPoints', help = 'number of corresponding points to input', default = 4, type = int) |
6551a3cf1750
modified compute-homography to work with argparse
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
443
diff
changeset
|
28 parser.add_argument('-u', dest = 'unitsPerPixel', help = 'number of units per pixel', default = 1., type = float) |
697
0421a5a0072c
improvemend by Adrien Lessard to choose output homography filename
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
640
diff
changeset
|
29 parser.add_argument('-o', dest = 'homographyFilename', help = 'filename of the homography matrix', default = 'homography.txt') |
476
6551a3cf1750
modified compute-homography to work with argparse
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
443
diff
changeset
|
30 parser.add_argument('--display', dest = 'displayPoints', help = 'display original and projected points on both images', action = 'store_true') |
510
b0dac840c24f
compute homography works with undistortion
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
478
diff
changeset
|
31 parser.add_argument('--intrinsic', dest = 'intrinsicCameraMatrixFilename', help = 'name of the intrinsic camera file') |
b0dac840c24f
compute homography works with undistortion
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
478
diff
changeset
|
32 parser.add_argument('--distortion-coefficients', dest = 'distortionCoefficients', help = 'distortion coefficients', nargs = '*', type = float) |
b0dac840c24f
compute homography works with undistortion
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
478
diff
changeset
|
33 parser.add_argument('--undistorted-multiplication', dest = 'undistortedImageMultiplication', help = 'undistorted image multiplication', type = float) |
538
bd1ad468e928
corrected bug and added capability to save undistorted image
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
510
diff
changeset
|
34 parser.add_argument('--undistort', dest = 'undistort', help = 'undistort the video (because features have been extracted that way', action = 'store_true') |
bd1ad468e928
corrected bug and added capability to save undistorted image
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
510
diff
changeset
|
35 parser.add_argument('--save', dest = 'saveImages', help = 'save the undistorted video frame (display option must be chosen)', action = 'store_true') |
476
6551a3cf1750
modified compute-homography to work with argparse
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
443
diff
changeset
|
36 |
6551a3cf1750
modified compute-homography to work with argparse
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
443
diff
changeset
|
37 args = parser.parse_args() |
151
4af774bb186d
wrote a simple script to compute homography from point correspondences and display the reprojection for visual verification
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
diff
changeset
|
38 |
217
ba71924cadf5
added comment/TODO
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
160
diff
changeset
|
39 # TODO process camera intrinsic and extrinsic parameters to obtain image to world homography, taking example from Work/src/python/generate-homography.py script |
ba71924cadf5
added comment/TODO
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
160
diff
changeset
|
40 # cameraMat = load(videoFilenamePrefix+'-camera.txt'); |
ba71924cadf5
added comment/TODO
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
160
diff
changeset
|
41 # T1 = cameraMat[3:6,:].copy(); |
ba71924cadf5
added comment/TODO
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
160
diff
changeset
|
42 # A = cameraMat[0:3,0:3].copy(); |
ba71924cadf5
added comment/TODO
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
160
diff
changeset
|
43 |
ba71924cadf5
added comment/TODO
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
160
diff
changeset
|
44 # # pay attention, rotation may be the transpose |
ba71924cadf5
added comment/TODO
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
160
diff
changeset
|
45 # # R = T1[:,0:3].T; |
ba71924cadf5
added comment/TODO
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
160
diff
changeset
|
46 # R = T1[:,0:3]; |
ba71924cadf5
added comment/TODO
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
160
diff
changeset
|
47 # rT = dot(R, T1[:,3]/1000); |
ba71924cadf5
added comment/TODO
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
160
diff
changeset
|
48 # T = zeros((3,4),'f'); |
ba71924cadf5
added comment/TODO
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
160
diff
changeset
|
49 # T[:,0:3] = R[:]; |
ba71924cadf5
added comment/TODO
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
160
diff
changeset
|
50 # T[:,3] = rT; |
ba71924cadf5
added comment/TODO
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
160
diff
changeset
|
51 |
ba71924cadf5
added comment/TODO
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
160
diff
changeset
|
52 # AT = dot(A,T); |
ba71924cadf5
added comment/TODO
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
160
diff
changeset
|
53 |
ba71924cadf5
added comment/TODO
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
160
diff
changeset
|
54 # nPoints = 4; |
ba71924cadf5
added comment/TODO
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
160
diff
changeset
|
55 # worldPoints = cvCreateMat(nPoints, 3, CV_64FC1); |
ba71924cadf5
added comment/TODO
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
160
diff
changeset
|
56 # imagePoints = cvCreateMat(nPoints, 3, CV_64FC1); |
ba71924cadf5
added comment/TODO
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
160
diff
changeset
|
57 |
ba71924cadf5
added comment/TODO
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
160
diff
changeset
|
58 # # extract homography from the camera calibration |
ba71924cadf5
added comment/TODO
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
160
diff
changeset
|
59 # worldPoints = cvCreateMat(4, 3, CV_64FC1); |
ba71924cadf5
added comment/TODO
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
160
diff
changeset
|
60 # imagePoints = cvCreateMat(4, 3, CV_64FC1); |
ba71924cadf5
added comment/TODO
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
160
diff
changeset
|
61 |
ba71924cadf5
added comment/TODO
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
160
diff
changeset
|
62 # worldPoints[0,:] = [[1, 1, 0]]; |
ba71924cadf5
added comment/TODO
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
160
diff
changeset
|
63 # worldPoints[1,:] = [[1, 2, 0]]; |
ba71924cadf5
added comment/TODO
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
160
diff
changeset
|
64 # worldPoints[2,:] = [[2, 1, 0]]; |
ba71924cadf5
added comment/TODO
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
160
diff
changeset
|
65 # worldPoints[3,:] = [[2, 2, 0]]; |
ba71924cadf5
added comment/TODO
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
160
diff
changeset
|
66 |
ba71924cadf5
added comment/TODO
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
160
diff
changeset
|
67 # wPoints = [[1,1,2,2], |
ba71924cadf5
added comment/TODO
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
160
diff
changeset
|
68 # [1,2,1,2], |
ba71924cadf5
added comment/TODO
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
160
diff
changeset
|
69 # [0,0,0,0]]; |
ba71924cadf5
added comment/TODO
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
160
diff
changeset
|
70 # iPoints = utils.worldToImage(AT, wPoints); |
ba71924cadf5
added comment/TODO
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
160
diff
changeset
|
71 |
ba71924cadf5
added comment/TODO
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
160
diff
changeset
|
72 # for i in range(nPoints): |
ba71924cadf5
added comment/TODO
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
160
diff
changeset
|
73 # imagePoints[i,:] = [iPoints[:,i].tolist()]; |
ba71924cadf5
added comment/TODO
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
160
diff
changeset
|
74 |
ba71924cadf5
added comment/TODO
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
160
diff
changeset
|
75 # H = cvCreateMat(3, 3, CV_64FC1); |
ba71924cadf5
added comment/TODO
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
160
diff
changeset
|
76 |
ba71924cadf5
added comment/TODO
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
160
diff
changeset
|
77 # cvFindHomography(imagePoints, worldPoints, H); |
ba71924cadf5
added comment/TODO
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
160
diff
changeset
|
78 |
151
4af774bb186d
wrote a simple script to compute homography from point correspondences and display the reprojection for visual verification
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
diff
changeset
|
79 |
238
be3761a09b20
added functions to input point correspondences
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
237
diff
changeset
|
80 homography = np.array([]) |
636
3058e00887bc
removed all issues because of tests with None, using is instead of == or !=
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
591
diff
changeset
|
81 if args.pointCorrespondencesFilename is not None: |
476
6551a3cf1750
modified compute-homography to work with argparse
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
443
diff
changeset
|
82 worldPts, videoPts = cvutils.loadPointCorrespondences(args.pointCorrespondencesFilename) |
238
be3761a09b20
added functions to input point correspondences
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
237
diff
changeset
|
83 homography, mask = cv2.findHomography(videoPts, worldPts) # method=0, ransacReprojThreshold=3 |
638
852f5de42d01
added functionality to read Aliaksei Tsai camera model data
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
636
diff
changeset
|
84 elif args.tsaiCameraFilename is not None: # hack using PDTV |
639
4e7925cb4f8f
modified tsai camera homography computation to avoid using os dependent temporary files
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
638
diff
changeset
|
85 from pdtv import TsaiCamera |
638
852f5de42d01
added functionality to read Aliaksei Tsai camera model data
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
636
diff
changeset
|
86 f = storage.openCheck(args.tsaiCameraFilename, quitting = True) |
852f5de42d01
added functionality to read Aliaksei Tsai camera model data
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
636
diff
changeset
|
87 content = storage.getLines(f) |
640
fe34c0f79c32
cleaner import that does not rely on StringIO or yaml
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
639
diff
changeset
|
88 cameraData = {} |
fe34c0f79c32
cleaner import that does not rely on StringIO or yaml
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
639
diff
changeset
|
89 for l in content: |
fe34c0f79c32
cleaner import that does not rely on StringIO or yaml
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
639
diff
changeset
|
90 tmp = l.split(':') |
fe34c0f79c32
cleaner import that does not rely on StringIO or yaml
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
639
diff
changeset
|
91 cameraData[tmp[0]] = float(tmp[1].strip().replace(',','.')) |
639
4e7925cb4f8f
modified tsai camera homography computation to avoid using os dependent temporary files
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
638
diff
changeset
|
92 camera = TsaiCamera(Cx=cameraData['Cx'], Cy=cameraData['Cy'], Sx=cameraData['Sx'], Tx=cameraData['Tx'], Ty=cameraData['Ty'], Tz=cameraData['Tz'], dx=cameraData['dx'], dy=cameraData['dy'], f=cameraData['f'], k=cameraData['k'], r1=cameraData['r1'], r2=cameraData['r2'], r3=cameraData['r3'], r4=cameraData['r4'], r5=cameraData['r5'], r6=cameraData['r6'], r7=cameraData['r7'], r8=cameraData['r8'], r9=cameraData['r9']) |
4e7925cb4f8f
modified tsai camera homography computation to avoid using os dependent temporary files
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
638
diff
changeset
|
93 homography = cvutils.computeHomographyFromPDTV(camera) |
636
3058e00887bc
removed all issues because of tests with None, using is instead of == or !=
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
591
diff
changeset
|
94 elif args.videoFrameFilename is not None and args.worldFilename is not None: |
476
6551a3cf1750
modified compute-homography to work with argparse
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
443
diff
changeset
|
95 worldImg = plt.imread(args.worldFilename) |
6551a3cf1750
modified compute-homography to work with argparse
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
443
diff
changeset
|
96 videoImg = plt.imread(args.videoFrameFilename) |
510
b0dac840c24f
compute homography works with undistortion
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
478
diff
changeset
|
97 if args.undistort: |
b0dac840c24f
compute homography works with undistortion
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
478
diff
changeset
|
98 [map1, map2] = cvutils.computeUndistortMaps(videoImg.shape[1], videoImg.shape[0], args.undistortedImageMultiplication, np.loadtxt(args.intrinsicCameraMatrixFilename), args.distortionCoefficients) |
b0dac840c24f
compute homography works with undistortion
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
478
diff
changeset
|
99 videoImg = cv2.remap(videoImg, map1, map2, interpolation=cv2.INTER_LINEAR) |
476
6551a3cf1750
modified compute-homography to work with argparse
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
443
diff
changeset
|
100 print('Click on {0} points in the video frame'.format(args.nPoints)) |
238
be3761a09b20
added functions to input point correspondences
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
237
diff
changeset
|
101 plt.figure() |
be3761a09b20
added functions to input point correspondences
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
237
diff
changeset
|
102 plt.imshow(videoImg) |
476
6551a3cf1750
modified compute-homography to work with argparse
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
443
diff
changeset
|
103 videoPts = np.array(plt.ginput(args.nPoints, timeout=3000)) |
6551a3cf1750
modified compute-homography to work with argparse
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
443
diff
changeset
|
104 print('Click on {0} points in the world image'.format(args.nPoints)) |
238
be3761a09b20
added functions to input point correspondences
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
237
diff
changeset
|
105 plt.figure() |
be3761a09b20
added functions to input point correspondences
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
237
diff
changeset
|
106 plt.imshow(worldImg) |
476
6551a3cf1750
modified compute-homography to work with argparse
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
443
diff
changeset
|
107 worldPts = args.unitsPerPixel*np.array(plt.ginput(args.nPoints, timeout=3000)) |
238
be3761a09b20
added functions to input point correspondences
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
237
diff
changeset
|
108 plt.close('all') |
be3761a09b20
added functions to input point correspondences
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
237
diff
changeset
|
109 homography, mask = cv2.findHomography(videoPts, worldPts) |
be3761a09b20
added functions to input point correspondences
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
237
diff
changeset
|
110 # save the points in file |
be3761a09b20
added functions to input point correspondences
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
237
diff
changeset
|
111 f = open('point-correspondences.txt', 'a') |
be3761a09b20
added functions to input point correspondences
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
237
diff
changeset
|
112 np.savetxt(f, worldPts.T) |
be3761a09b20
added functions to input point correspondences
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
237
diff
changeset
|
113 np.savetxt(f, videoPts.T) |
be3761a09b20
added functions to input point correspondences
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
237
diff
changeset
|
114 f.close() |
151
4af774bb186d
wrote a simple script to compute homography from point correspondences and display the reprojection for visual verification
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
diff
changeset
|
115 |
238
be3761a09b20
added functions to input point correspondences
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
237
diff
changeset
|
116 if homography.size>0: |
697
0421a5a0072c
improvemend by Adrien Lessard to choose output homography filename
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
640
diff
changeset
|
117 np.savetxt(args.homographyFilename,homography) |
238
be3761a09b20
added functions to input point correspondences
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
237
diff
changeset
|
118 |
638
852f5de42d01
added functionality to read Aliaksei Tsai camera model data
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
636
diff
changeset
|
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: |
478
d337bffd7283
Display of points in compute homography and step option to replay videos
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
476
diff
changeset
|
120 worldImg = cv2.imread(args.worldFilename) |
d337bffd7283
Display of points in compute homography and step option to replay videos
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
476
diff
changeset
|
121 videoImg = cv2.imread(args.videoFrameFilename) |
572
9c429c7efe89
added script to generate undistorted images from video
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
538
diff
changeset
|
122 if args.undistort: |
510
b0dac840c24f
compute homography works with undistortion
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
478
diff
changeset
|
123 [map1, map2] = cvutils.computeUndistortMaps(videoImg.shape[1], videoImg.shape[0], args.undistortedImageMultiplication, np.loadtxt(args.intrinsicCameraMatrixFilename), args.distortionCoefficients) |
b0dac840c24f
compute homography works with undistortion
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
478
diff
changeset
|
124 videoImg = cv2.remap(videoImg, map1, map2, interpolation=cv2.INTER_LINEAR) |
538
bd1ad468e928
corrected bug and added capability to save undistorted image
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
510
diff
changeset
|
125 if args.saveImages: |
bd1ad468e928
corrected bug and added capability to save undistorted image
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
510
diff
changeset
|
126 cv2.imwrite(utils.removeExtension(args.videoFrameFilename)+'-undistorted.png', videoImg) |
151
4af774bb186d
wrote a simple script to compute homography from point correspondences and display the reprojection for visual verification
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
diff
changeset
|
127 invHomography = np.linalg.inv(homography) |
238
be3761a09b20
added functions to input point correspondences
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
237
diff
changeset
|
128 projectedWorldPts = cvutils.projectArray(invHomography, worldPts.T).T |
478
d337bffd7283
Display of points in compute homography and step option to replay videos
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
476
diff
changeset
|
129 projectedVideoPts = cvutils.projectArray(homography, videoPts.T).T |
238
be3761a09b20
added functions to input point correspondences
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
237
diff
changeset
|
130 for i in range(worldPts.shape[0]): |
478
d337bffd7283
Display of points in compute homography and step option to replay videos
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
476
diff
changeset
|
131 # world image |
d337bffd7283
Display of points in compute homography and step option to replay videos
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
476
diff
changeset
|
132 cv2.circle(worldImg,tuple(np.int32(np.round(worldPts[i]/args.unitsPerPixel))),2,cvutils.cvBlue) |
476
6551a3cf1750
modified compute-homography to work with argparse
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
443
diff
changeset
|
133 cv2.circle(worldImg,tuple(np.int32(np.round(projectedVideoPts[i]/args.unitsPerPixel))),2,cvutils.cvRed) |
478
d337bffd7283
Display of points in compute homography and step option to replay videos
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
476
diff
changeset
|
134 cv2.putText(worldImg, str(i+1), tuple(np.int32(np.round(worldPts[i]/args.unitsPerPixel))+5), cv2.FONT_HERSHEY_PLAIN, 2., cvutils.cvBlue, 2) |
d337bffd7283
Display of points in compute homography and step option to replay videos
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
476
diff
changeset
|
135 # video image |
d337bffd7283
Display of points in compute homography and step option to replay videos
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
476
diff
changeset
|
136 cv2.circle(videoImg,tuple(np.int32(np.round(videoPts[i]))),2,cvutils.cvBlue) |
d337bffd7283
Display of points in compute homography and step option to replay videos
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
476
diff
changeset
|
137 cv2.circle(videoImg,tuple(np.int32(np.round(projectedWorldPts[i]))),2,cvutils.cvRed) |
d337bffd7283
Display of points in compute homography and step option to replay videos
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
476
diff
changeset
|
138 cv2.putText(videoImg, str(i+1), tuple(np.int32(np.round(videoPts[i])+5)), cv2.FONT_HERSHEY_PLAIN, 2., cvutils.cvBlue, 2) |
238
be3761a09b20
added functions to input point correspondences
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
237
diff
changeset
|
139 cv2.imshow('video frame',videoImg) |
478
d337bffd7283
Display of points in compute homography and step option to replay videos
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
476
diff
changeset
|
140 cv2.imshow('world image',worldImg) |
151
4af774bb186d
wrote a simple script to compute homography from point correspondences and display the reprojection for visual verification
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
diff
changeset
|
141 cv2.waitKey() |
346
5f75d6c23ed5
added opencv function to destroy OpenCV windows (seems to work only on Windows)
Nicolas Saunier <nicolas.saunier@polymtl.ca>
parents:
334
diff
changeset
|
142 cv2.destroyAllWindows() |