Mercurial Hosting > traffic-intelligence

comparison scripts/compute-homography.py @ 868:1fdafa9f6bf4

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
added colors more friendly for color blind people (thanks Ryan Louie!)
author Nicolas Saunier <nicolas.saunier@polymtl.ca>
date Wed, 01 Feb 2017 11:58:04 -0500
parents bf4a1790cfac
children 739acd338cc0
comparison
equal deleted inserted replaced
867:003445db1e30 868:1fdafa9f6bf4
129 invHomography = np.linalg.inv(homography) 129 invHomography = np.linalg.inv(homography)
130 projectedWorldPts = cvutils.projectArray(invHomography, worldPts.T).T 130 projectedWorldPts = cvutils.projectArray(invHomography, worldPts.T).T
131 projectedVideoPts = cvutils.projectArray(homography, videoPts.T).T 131 projectedVideoPts = cvutils.projectArray(homography, videoPts.T).T
132 for i in range(worldPts.shape[0]): 132 for i in range(worldPts.shape[0]):
133 # world image 133 # world image
134 cv2.circle(worldImg,tuple(np.int32(np.round(worldPts[i]/args.unitsPerPixel))),2,cvutils.cvBlue) 134 cv2.circle(worldImg,tuple(np.int32(np.round(worldPts[i]/args.unitsPerPixel))),2,cvutils.cvBlue['default'])
135 cv2.circle(worldImg,tuple(np.int32(np.round(projectedVideoPts[i]/args.unitsPerPixel))),2,cvutils.cvRed) 135 cv2.circle(worldImg,tuple(np.int32(np.round(projectedVideoPts[i]/args.unitsPerPixel))),2,cvutils.cvRed['default'])
136 cv2.putText(worldImg, str(i+1), tuple(np.int32(np.round(worldPts[i]/args.unitsPerPixel))+5), cv2.FONT_HERSHEY_PLAIN, 2., cvutils.cvBlue, 2) 136 cv2.putText(worldImg, str(i+1), tuple(np.int32(np.round(worldPts[i]/args.unitsPerPixel))+5), cv2.FONT_HERSHEY_PLAIN, 2., cvutils.cvBlue['default'], 2)
137 # video image 137 # video image
138 cv2.circle(videoImg,tuple(np.int32(np.round(videoPts[i]))),2,cvutils.cvBlue) 138 cv2.circle(videoImg,tuple(np.int32(np.round(videoPts[i]))),2,cvutils.cvBlue['default'])
139 cv2.circle(videoImg,tuple(np.int32(np.round(projectedWorldPts[i]))),2,cvutils.cvRed) 139 cv2.circle(videoImg,tuple(np.int32(np.round(projectedWorldPts[i]))),2,cvutils.cvRed['default'])
140 cv2.putText(videoImg, str(i+1), tuple(np.int32(np.round(videoPts[i])+5)), cv2.FONT_HERSHEY_PLAIN, 2., cvutils.cvBlue, 2) 140 cv2.putText(videoImg, str(i+1), tuple(np.int32(np.round(videoPts[i])+5)), cv2.FONT_HERSHEY_PLAIN, 2., cvutils.cvBlue['default'], 2)
141 cv2.imshow('video frame',videoImg) 141 cv2.imshow('video frame',videoImg)
142 cv2.imshow('world image',worldImg) 142 cv2.imshow('world image',worldImg)
143 cv2.waitKey() 143 cv2.waitKey()
144 cv2.destroyAllWindows() 144 cv2.destroyAllWindows()