1獲取普通USB相機
#-*- coding: UTF-8 -*-
import numpy as np # 引入numpy 用於矩陣運算
import cv2 # 引入opencv庫函數
## VideCapture里面的序號
# 0 : 默認為筆記本上的攝像頭(如果有的話) / USB攝像頭 webcam
# 1 : USB攝像頭2
# 2 :USB攝像頭3 以此類推
# -1:代表最新插入的USB設備
# 創建一個video capture的實例
cap = cv2.VideoCapture(0)
# 查看Video Capture是否已經打開
print("攝像頭是否已經打開 ? {}".format(cap.isOpened()))
## 設置畫面的尺寸
# 畫面寬度設定為 1920
cap.set(cv2.CAP_PROP_FRAME_WIDTH, 1920)
# 畫面高度度設定為 1080
cap.set(cv2.CAP_PROP_FRAME_HEIGHT, 1080)
## 創建一個名字叫做 “image_win” 的窗口
# 窗口屬性 flags
# * WINDOW_NORMAL:窗口可以放縮
# * WINDOW_KEEPRATIO:窗口縮放的過程中保持比率
# * WINDOW_GUI_EXPANDED: 使用新版本功能增強的GUI窗口
cv2.namedWindow('image_win',flags=cv2.WINDOW_NORMAL | cv2.WINDOW_KEEPRATIO | cv2.WINDOW_GUI_EXPANDED)
# 圖像計數 從1開始
img_count = 1
# 幫助信息
helpInfo = '''
=======阿凱貼心小助手=======
提示-按鍵前需要選中當前畫面顯示的窗口
按鍵Q: 退出程序
按鍵C: Capture 拍照
'''
print(helpInfo)
while(True):
## 逐幀獲取畫面
# 如果畫面讀取成功 ret=True,frame是讀取到的圖片對象(numpy的ndarray格式)
ret, frame = cap.read()
if not ret:
# 如果圖片沒有讀取成功
print("圖像獲取失敗,請按照說明進行問題排查")
## 讀取失敗?問題排查
# **驅動問題** 有的攝像頭可能存在驅動問題,需要安裝相關驅動,或者查看攝像頭是否有UVC免驅協議
# **接口兼容性問題** 或者USB2.0接口接了一個USB3.0的攝像頭,也是不支持的。
# **設備掛載問題** 攝像頭沒有被掛載,如果是虛擬機需要手動勾選設備
# **硬件問題** 在就是檢查一下USB線跟電腦USB接口
break
## 顏色空間變換
# 將BGR彩圖變換為灰度圖
# frame = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
## 圖片鏡像
# * 水平翻轉 flipCode = 1
# * 垂直翻轉 flipCode = 0
# * 同時水平翻轉與垂直翻轉 flipCode = -1
#
# flipCode = -1
# frame = cv2.flip(frame, flipCode)
# 更新窗口“image_win”中的圖片
cv2.imshow('image_win',frame)
# 等待按鍵事件發生 等待1ms
key = cv2.waitKey(1)
if key == ord('q'):
# 如果按鍵為q 代表quit 退出程序
print("程序正常退出...Bye 不要想我哦")
break
elif key == ord('c'):
## 如果c鍵按下,則進行圖片保存
# 寫入圖片 並命名圖片為 圖片序號.png
cv2.imwrite("{}.png".format(img_count), frame)
print("截圖,並保存為 {}.png".format(img_count))
# 圖片編號計數自增1
img_count += 1
# 釋放VideoCapture
cap.release()
# 銷毀所有的窗口
cv2.destroyAllWindows()
2獲取樹莓派官方攝像頭

sudo raspi-config
使用箭頭鍵向下滾動到選項5:啟用相機,按回車鍵啟用相機,然后向下箭頭至Finish按鈕,再次按Enter。最后,你需要重新啟動你的 Raspberry Pi 才能使配置生效。

第三步:測試相機模塊

raspistill -o output.jpg

第四步:安裝 picamera
pip install "picamera[array]"
注意:我指定了 picamera[array] 而不是 picamera。
當操作 picamera 模塊中的方法來和相機交互時,如果我們需要使用 OpenCV,我們需要使用 array 子模塊。在 Python 中,OpenCV 的圖像格式為 Numpy arrays——而 array 子模塊允許我從樹莓派相機模塊中獲取 Numpy arrays。
第五步:使用 Python 和 OpenCV 訪問 Raspberry Pi 的單個圖像
現在,開始編寫代碼!
# import the necessary packages
from picamera.array import PiRGBArray
from picamera import PiCamera
import time
import cv2
# initialize the camera and grab a reference to the raw camera capture
camera = PiCamera()
rawCapture = PiRGBArray(camera)
# allow the camera to warmup
time.sleep(0.1)
# grab an image from the camera
camera.capture(rawCapture, format="bgr")
image = rawCapture.array
# display the image on screen and wait for a keypress
cv2.imshow("Image", image)
cv2.waitKey(0)
第2-5行代碼表示導入相關的包。
第8行代碼表示初始化 PiCamera 對象,第9行代碼表示獲取對原始捕獲組件的引用。 這個 rawCapture 對象特別有用,因為:
- 可以直接訪問相機流。
- 避免了壓縮流為 JPEG 格式的時間,因為我們最終得到的是 OpenCV 圖像格式。
我強烈推薦在樹莓派相機上使用PiRGBArray函數——因為它的性能值得一試。
接下來在第12行,我們 sleep 2秒——為了預熱相機模塊,必須這樣做。
第15行我們從相機上捕獲了一張圖片,保存在 rawCapture 對象上,並且明確圖像顏色排列為 BGR 而不是 RGB——這是因為 OpenCV 的圖像在 Python 中的格式為 BGR 而不是 RGB。這很重要,不注意這個很可能會產生錯誤的結果。
最后,第19和第20行展示捕獲了的圖片。
第六步:使用 Python 和 OpenCV 訪問 Raspberry Pi 的視頻流
好的,現在我們已經學會了如何從 Raspberry Pi 相機中獲取單個圖像。那么視頻流呢?
你可能會猜想,我們將在這里使用 cv2.VideoCapture 函數——但實際上我建議不要這樣做。 把 cv2.VideoCapture 與 Raspberry Pi 配合使用並不是一個好的體驗(你需要安裝額外的驅動程序),而你通常應該會避免使用它。 另外,當我們可以使用 picamera 模塊輕松訪問原始視頻流時,為什么我們還要去使用 cv2.VideoCapture 函數呢?
讓我們繼續看看如何訪問視頻流。打開一個新文件,命名為 test_video.py,然后插入以下代碼:
# import the necessary packages
from picamera.array import PiRGBArray
from picamera import PiCamera
import time
import cv2
# initialize the camera and grab a reference to the raw camera capture
camera = PiCamera()
camera.resolution = (640, 480)
camera.framerate = 32
rawCapture = PiRGBArray(camera, size=(640, 480))
# allow the camera to warmup
time.sleep(0.1)
# capture frames from the camera
for frame in camera.capture_continuous(rawCapture, format="bgr", use_video_port=True):
# grab the raw NumPy array representing the image, then initialize the timestamp
# and occupied/unoccupied text
image = frame.array
# show the frame
cv2.imshow("Frame", image)
key = cv2.waitKey(1) & 0xFF
# clear the stream in preparation for the next frame
rawCapture.truncate(0)
# if the `q` key was pressed, break from the loop
if key == ord("q"):
break
正如你所看到的,Raspberry Pi 相機的視頻流正在被 OpenCV 讀取,然后顯示在屏幕上!此外,Raspberry Pi 相機在以 32 FPS 的速度訪問畫面時不會出現滯后現象。 當然,我們並沒有對單個幀進行任何處理,但是正如我在博客文章中所展示的那樣,即使在處理每個幀時,Pi 2 也可以很容易地保持 24-32 FPS。
