2012-06-28 9 views
5

Ho un programma che salva un'immagine in una directory locale e quindi legge l'immagine da quella directory.Python: Urllib2 e OpenCV

Ma non voglio salvare l'immagine. Voglio leggerlo direttamente dall'URL.

Ecco il mio codice:

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) 

Se cambio a:

img = cv.LoadImage(img) 

Questo mi darà questo errore:

argument 1 must be string without null bytes, not str

Cosa posso fare?

risposta

4

Se si vuole si può utilizzare 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) 

Immagino che con questo metodo non sia necessario salvare il file di immagine.

+0

Grazie per questo, molto utile! Un punto: ho ricevuto un errore con "da CStringIO ...". Dovrebbe essere "da cStringIO ..." (in minuscolo 'c'). –

+0

mio male. Grazie per la segnalazione. L'ho cambiato. – Froyo

-1

Come accennato here LoadImage attesa nome del file come primo argomento, non i dati