samples/python/cvutils.py

   1 #!/usr/bin/python
   2
   3 import cv2.cv as cv
   4 import urllib2
   5 from sys import argv
   6
   7 def load_sample(name=None):
   8     if len(argv) > 1:
   9         img0 = cv.LoadImage(argv[1], cv.CV_LOAD_IMAGE_COLOR)
  10     elif name is not None:
  11         try:
  12             img0 = cv.LoadImage(name, cv.CV_LOAD_IMAGE_COLOR)
  13         except IOError:
  14             urlbase = 'http://code.opencv.org/projects/opencv/repository/revisions/master/raw/samples/c/'
  15             file = name.split('/')[-1]
  16             filedata = urllib2.urlopen(urlbase+file).read()
  17             imagefiledata = cv.CreateMatHeader(1, len(filedata), cv.CV_8UC1)
  18             cv.SetData(imagefiledata, filedata, len(filedata))
  19             img0 = cv.DecodeImage(imagefiledata, cv.CV_LOAD_IMAGE_COLOR)
  20     return img0