2012-06-28 3 views
5

У меня есть программа, которая сохраняет изображение в локальном каталоге и затем считывает изображение из этого каталога.Python: Urllib2 и OpenCV

Но я не хочу сохранять изображение. Я хочу прочитать его прямо из URL-адреса.

Вот мой код:

import cv2.cv as cv 
import urllib2 

url = "http://cache2.allpostersimages.com/p/LRG/18/1847/M5G8D00Z/posters/curious-cat.jpg" 
filename = "my_test_image" + url[-4:] 
print filename 
opener = urllib2.build_opener() 

page = opener.open(url) 
img= page.read() 

abc = open(filename, "wb") 
abc.write(img) 
abc.close() 

img = cv.LoadImage(filename) 

cv.ShowImage("Optical Flow", img) 
cv.WaitKey(30) 

Если я изменить его на:

img = cv.LoadImage(img) 

Это даст мне эту ошибку:

argument 1 must be string without null bytes, not str

Что я могу сделать?

ответ

4

Если вы хотите, вы можете использовать PIL.

import cv2.cv as cv 
import urllib2 
from cStringIO import StringIO 
import PIL.Image as pil 
url="some_url" 

img_file = urllib2.urlopen(url) 
im = StringIO(img_file.read()) 
source = pil.open(im).convert("RGB") 
bitmap = cv.CreateImageHeader(source.size, cv.IPL_DEPTH_8U, 3) 
cv.SetData(bitmap, source.tostring()) 
cv.CvtColor(bitmap, bitmap, cv.CV_RGB2BGR) 

Я думаю, что этим способом вам не нужно сохранять файл изображения.

+0

Спасибо за это, очень полезно! Один момент: у меня возникла ошибка с «из CStringIO ...». Это должно быть «from cStringIO ...» (нижний регистр «c»). –

+0

мой плохой. Спасибо что подметил это. Я изменил его. – Froyo

-1

Как уже упоминалось here LoadImage ожидает имя файла в качестве первого аргумента, а не данных