快捷導(dǎo)航

利用Python和OpenCV庫(kù)將URL轉(zhuǎn)換為OpenCV格式的方法

更新時(shí)間：2015年03月27日 15:08:40 作者：Adrian Rosebrock

這篇文章主要介紹了利用Python和OpenCV庫(kù)將URL轉(zhuǎn)換為OpenCV格式的方法,同時(shí)用到了NumPy和urllib,需要的朋友可以參考下

2015327143841977.jpg (690×553)

今天的博客是直接來(lái)源于我自己的個(gè)人工具函數(shù)庫(kù)。

過(guò)去幾個(gè)月，有些PyImageSearch讀者電郵問我：“如何獲取URL指向的圖片并將其轉(zhuǎn)換成OpenCV格式（不用將其寫入磁盤再讀回）”。這篇文章我將展示一下怎么實(shí)現(xiàn)這個(gè)功能。

額外的，我們也會(huì)看到如何利用scikit-image從URL下載一幅圖像。當(dāng)然前行之路也會(huì)有一個(gè)常見的錯(cuò)誤，它可能讓你跌個(gè)跟頭。

繼續(xù)往下閱讀，學(xué)習(xí)如何利用利用Python和OpenCV將URL轉(zhuǎn)換為圖像

方法1：OpenCV、NumPy、urllib

第一個(gè)方法：我們使用OpenCV、NumPy、urllib庫(kù)從URL獲取圖像，并將其轉(zhuǎn)換為圖像。打開并新建一個(gè)文件，取名url_to_image.py，我們開始吧：

# import the necessary packages
import numpy as np
import urllib
import cv2
 
# METHOD #1: OpenCV, NumPy, and urllib
def url_to_image(url):
  # download the image, convert it to a NumPy array, and then read
  # it into OpenCV format
  resp = urllib.urlopen(url)
  image = np.asarray(bytearray(resp.read()), dtype="uint8")
  image = cv2.imdecode(image, cv2.IMREAD_COLOR)
 
  # return the image
  return image

首先要做的就是導(dǎo)入我們必需的包。我們將使用NumPy轉(zhuǎn)換下載的字節(jié)序?yàn)镹umPy數(shù)組，使用urllib來(lái)執(zhí)行實(shí)際的網(wǎng)絡(luò)請(qǐng)求，使用cv2來(lái)綁定OpenCV接口。
在第7行，我們定義了我們的url_to_image函數(shù)。這個(gè)函數(shù)帶一個(gè)url參數(shù)，也就是我們想要下載的圖像地址。

接下來(lái)，在第10行，我們使用urllib庫(kù)來(lái)打開這個(gè)圖像鏈接。11行則將這個(gè)下載下來(lái)的字節(jié)序轉(zhuǎn)換為NumPy數(shù)組。

至此，NumPy數(shù)組還是一個(gè)1維數(shù)組（也就是一個(gè)長(zhǎng)長(zhǎng)的像素鏈表）。為了將其轉(zhuǎn)換為2維格式，假設(shè)每個(gè)像素3個(gè)通道（意即分別為紅，綠，藍(lán)通道），在12行我們使用cv.imdecode函數(shù)。最后，在15行我們返回解碼出來(lái)的圖像給調(diào)用函數(shù)。

一切就緒，該到讓它工作的時(shí)候了：

# initialize the list of image URLs to download
urls = [
  "http://www.pyimagesearch.com/wp-content/uploads/2015/01/opencv_logo.png",
  "http://www.pyimagesearch.com/wp-content/uploads/2015/01/google_logo.png",
  "http://www.pyimagesearch.com/wp-content/uploads/2014/12/adrian_face_detection_sidebar.png",
]
 
# loop over the image URLs
for url in urls:
  # download the image URL and display it
  print "downloading %s" % (url)
  image = url_to_image(url)
  cv2.imshow("Image", image)
  cv2.waitKey(0)

3-5行定義了我們將要下載和轉(zhuǎn)換為OpenCV格式的圖像地址列表。

第9行我們遍歷這個(gè)列表，13行則調(diào)用url_to_image函數(shù)，然后在14行和15行將獲取的圖像顯示到屏幕上。到此呢，我們就可以像正常情況下一樣，使用OpenCV來(lái)操作和處理這些圖像了。

眼見為實(shí)，打開終端，執(zhí)行如下指令：

復(fù)制代碼代碼如下:

$ python url_to_image.py

如果一切順利的話，你會(huì)看到OpenCV的logo：

2015327144145476.jpg (690×553)

圖1：從URL下載OpenCV logo并轉(zhuǎn)換為OpenCV格式

接下來(lái)是Google的logo：

2015327144648394.jpg (690×296)

圖2：從URL下載Gooogle并轉(zhuǎn)換為OpenCV格式

這里也有在我書中驗(yàn)證人臉檢測(cè)的例子，《Practical Python and OpenCV》:

2015327145507830.jpg (690×292)

圖3：轉(zhuǎn)換一個(gè)URL圖像為OpenCV格式

現(xiàn)在，我們來(lái)看另一種獲取圖像并轉(zhuǎn)換為OpenCV格式的方法。

方法2：使用scikit-image

第二種方法假定你已經(jīng)在你計(jì)算機(jī)上安裝好了scikit-image庫(kù)。讓我們看看怎樣采用scikit-image從URL獲取圖像并將其轉(zhuǎn)換為OpenCV格式：

# METHOD #2: scikit-image
from skimage import io
 
# loop over the image URLs
for url in urls:
  # download the image using scikit-image
  print "downloading %s" % (url)
  image = io.imread(url)
  cv2.imshow("Incorrect", image)
  cv2.imshow("Correct", cv2.cvtColor(image, cv2.COLOR_BGR2RGB))
  cv2.waitKey(0)

scikit-image庫(kù)中做得很漂亮的一點(diǎn)是：io子庫(kù)中的imread函數(shù)能夠區(qū)分圖像路徑到底在磁盤上還是一個(gè)URL（第9行)。

盡管這樣，這里有一個(gè)很嚴(yán)重的錯(cuò)誤可能讓你跌一個(gè)跟頭！

OpenCV以BGR順序表達(dá)一幅圖像，然而scikit-image則是RGB順序。如果你使用scikit-iamge的imread函數(shù)，而且還想在下載完成后使用OpenCV的函數(shù)，那么你要小心了。如41行所述，你需要將圖像從RBG轉(zhuǎn)換為BGR。

如果你沒有這一步，那么你可能得到錯(cuò)誤的結(jié)果：

2015327145042117.jpg (690×368)