Python Cookbook - Opencv v2

Cookbook
Here is a collection of code fragments demonstrating some features of the OpenCV Python bind
Convert an image
>>> import cv >>> im = cv.LoadImageM("building.jpg") >>> print type(im) <type 'cv.cvmat'> >>> cv.SaveImage("foo.png", im)
Resize an image
To resize an image in OpenCV, create a destination image of the appropriate size, then call Re
>>> >>> >>> >>> import cv original = cv.LoadImageM("building.jpg") thumbnail = cv.CreateMat(original.rows / 10, original.cols / 10, cv.CV_8UC3) cv.Resize(original, thumbnail)
Compute the Laplacian

>>> >>> >>> >>> >>> import cv im = cv.LoadImageM("building.jpg", 1) dst = cv.CreateImage(cv.GetSize(im), cv.IPL_DEPTH_16S, 3) laplace = cv.Laplace(im, dst) cv.SaveImage("foo-laplace.png", dst)
Using GoodFeaturesToTrack
To find the 10 strongest corner features in an image, use GoodFeaturesToTrack like this:
>>> import cv >>> img = cv.LoadImageM("building.jpg", cv.CV_LOAD_IMAGE_GRAYSCALE) >>> eig_image = cv.CreateMat(img.rows, img.cols, cv.CV_32FC1) >>> temp_image = cv.CreateMat(img.rows, img.cols, cv.CV_32FC1) >>> for (x,y) in cv.GoodFeaturesToTrack(img, eig_image, temp_image, 10, 0.04, 1.0 ... print "good feature at", x,y good feature at 198.0 514.0 good feature at 791.0 260.0 good feature at 370.0 467.0 good feature at 374.0 469.0 good feature at 490.0 520.0 good feature at 262.0 278.0 good feature at 781.0 134.0 good feature at 3.0 247.0 good feature at 667.0 321.0 good feature at 764.0 304.0
Using GetSubRect
GetSubRect returns a rectangular part of another image. It does this without copying any data.
>>> >>> >>> >>> import cv img = cv.LoadImageM("building.jpg") sub = cv.GetSubRect(img, (60, 70, 32, 32)) cv.SetZero(sub)
# sub is 32x32 patch within img # clear sub to zero, which also cl
Using CreateMat, and accessing an element
>>> import cv >>> mat = cv.CreateMat(5, 5, cv.CV_32FC1) >>> cv.Set(mat, 1.0) >>> mat[3,1] += 0.375 >>> print mat[3,1] 1.375 >>> print [mat[3,i] for i in range(5)] [1.0, 1.375, 1.0, 1.0, 1.0]
ROS image message to OpenCV

See this tutorial: Using CvBridge to convert between ROS images And OpenCV images .
PIL Image to OpenCV

(For details on PIL see the PIL handbook .)
>>> import Image, cv # PIL image >>> pi = Image.open('building.jpg') >>> cv_im = cv.CreateImageHeader(pi.size, cv.IPL_DEPTH_8U, 3) >>> cv.SetData(cv_im, pi.tostring()) >>> print pi.size, cv.GetSize(cv_im) (868, 600) (868, 600) >>> print pi.tostring() == cv_im.tostring() True
OpenCV to PIL Image

>>> import Image, cv >>> cv_im = cv.CreateImage((320,200), cv.IPL_DEPTH_8U, 1) >>> pi = Image.fromstring("L", cv.GetSize(cv_im), cv_im.tostring()) >>> print pi.size (320, 200)
NumPy and OpenCV

Using the array interface , to use an OpenCV CvMat in NumPy:
>>> import cv, numpy >>> mat = cv.CreateMat(3, 5, cv.CV_32FC1) >>> cv.Set(mat, 7) >>> a = numpy.asarray(mat) >>> print a [[ 7. 7. 7. 7. 7.] [ 7. 7. 7. 7. 7.] [ 7. 7. 7. 7. 7.]]
and to use a NumPy array in OpenCV:

>>> >>> >>> >>> 480 >>> 640 import cv, numpy a = numpy.ones((480, 640)) mat = cv.fromarray(a) print mat.rows print mat.cols
also, most OpenCV functions can work on NumPy arrays directly, for example:
>>> picture = numpy.ones((640, 480)) >>> cv.Smooth(picture, picture, cv.CV_GAUSSIAN, 15, 15)
Given a 2D array, the fromarray function (or the implicit version shown above) returns a single size. For a 3D array of size , it returns a CvMat sized with channels. Alternatively, use fromarray with the allowND option to always return a CvMatND .
OpenCV to pygame
To convert an OpenCV image to a pygame surface:

>>> import pygame.image, cv >>> src = cv.LoadImage("lena.jpg") >>> src_rgb = cv.CreateMat(src.height, src.width, cv.CV_8UC3) >>> cv.CvtColor(src, src_rgb, cv.CV_BGR2RGB) >>> pg_img = pygame.image.frombuffer(src_rgb.tostring(), cv.GetSize(src_rgb), "RGB >>> print pg_img <Surface(512x512x24 SW)>
OpenCV and OpenEXR

Using OpenEXRs Python bindings you can make a simple image viewer:
import OpenEXR, Imath, cv filename = "GoldenGate.exr" exrimage = OpenEXR.InputFile(filename) dw = exrimage.header()['dataWindow'] (width, height) = (dw.max.x - dw.min.x + 1, dw.max.y - dw.min.y + 1) def fromstr(s): mat = cv.CreateMat(height, width, cv.CV_32FC1) cv.SetData(mat, s) return mat pt = Imath.PixelType(Imath.PixelType.FLOAT) (r, g, b) = [fromstr(s) for s in exrimage.channels("RGB", pt)] bgr = cv.CreateMat(height, width, cv.CV_32FC3) cv.Merge(b, g, r, None, bgr) cv.ShowImage(filename, bgr) cv.WaitKey()
Help and Feedback

You did not find what you were looking for? Try the Cookbook. Ask a question in the user group/mailing list. If you think something is missing or wrong in the documentation, please file a bug report.

Python Cookbook - Opencv v2

Uploaded by

Python Cookbook - Opencv v2

Uploaded by

Cookbook

Compute the Laplacian

Using CreateMat, and accessing an element

ROS image message to OpenCV

PIL Image to OpenCV

OpenCV to PIL Image

NumPy and OpenCV

and to use a NumPy array in OpenCV:

To convert an OpenCV image to a pygame surface:

OpenCV and OpenEXR

Help and Feedback

You might also like