1. 程式人生 > >Python OpenCV——基本操作一網打盡

Python OpenCV——基本操作一網打盡

  • cv2.imread()——讀取影象

    程式碼格式cv2.imread(img,flag)
    flag取值可以為
    * cv2.IMREAD_COLOR 預設、彩色照片
    * cv2.IMREAD_GRAYSCALE 灰度照片
    * cv2.IMREAD_UNCHANGED 載入原圖通道
    也可以通過1、 0、-1 指定上述三個模式
import numpy as np
import cv2

# Load an color image in grayscale
img = cv2.imread('messi5.jpg',0)
  • cv2.imshow()——顯示圖片
cv2.imshow('image',img)#第一個為顯示的圖片名字,第二個為圖片
cv2.waitKey(0)#不停重新整理圖片,直到你按下任意按鍵退出
cv2.destroyAllWindows()#關閉所有顯示框,若只關閉一個,可 cv2.destroyWindow()單獨指定
  • cv2.imwrite()——儲存圖片
cv2.imwrite('messigray.png',img) #第一個引數為檔名,第二個為要儲存的影象。
  • cv2.VideoCapture()——捕捉視訊影象
import numpy as np
import cv2

#建立一個VideoCapture專案,0代表選擇第一個裝置
cap = cv2.VideoCapture(0) while(True): # 捕捉每一幀影象,返回兩個引數ret為是否返回圖片(True or False),frame為返回的圖片 ret, frame = cap.read() # 通過cv2.cvtColor轉換顏色格式為灰度模式 gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY) # 播放每一幀影象 cv2.imshow('frame',gray) if cv2.waitKey(1) & 0xFF == ord('q'): break
# 最後要記得釋放capture cap.release() cv2.destroyAllWindows()

還可以通過更改裝置號為檔名來播放視訊,並且通過cv2.WaitKey()來控制播放速度,可以製作慢動作視訊等等

import numpy as np
import cv2

cap = cv2.VideoCapture('vtest.avi')

while(cap.isOpened()):
    ret, frame = cap.read()

    gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)

    cv2.imshow('frame',gray)
    if cv2.waitKey(1) & 0xFF == ord('q'):
        break

cap.release()
cv2.destroyAllWindows()
  • cv2.VideoWriter()——儲存視訊
import numpy as np
import cv2

cap = cv2.VideoCapture(0)

#fourcc是一種編碼格式,我們儲存視訊時要指定檔名、編碼格式、FPS、輸出尺寸、顏色模式
fourcc = cv2.VideoWriter_fourcc(*'XVID')
out = cv2.VideoWriter('output.avi',fourcc, 20.0, (640,480))

while(cap.isOpened()):
    ret, frame = cap.read()
    if ret==True:
        #cv2.flip(img,flag)  翻轉影象(1水平翻轉、0垂直翻轉、-1水平垂直翻轉)
        frame = cv2.flip(frame,0)

        # write the flipped frame
        out.write(frame)

        cv2.imshow('frame',frame)
        if cv2.waitKey(1) & 0xFF == ord('q'):
            break
    else:
        break

cap.release()
out.release()
cv2.destroyAllWindows()
  • cv2.line(), cv2.circle() , cv2.rectangle(), cv2.ellipse(), cv2.putText()
    畫圖,用處較少,不再贅述
  • cv2.setMouseCallback()——滑鼠繪圖
import cv2
import numpy as np

#初始化
drawing = False #為真時開始畫圖
mode = True #為真時畫舉行,為假時畫圓
ix,iy = -1,-1


def draw_circle(event,x,y,flags,param):
    global ix,iy,drawing,mode

    #滑鼠左鍵按下時
    if event == cv2.EVENT_LBUTTONDOWN:
        drawing = True
        ix,iy = x,y

    #滑鼠移動時
    elif event == cv2.EVENT_MOUSEMOVE:
        if drawing == True:
            if mode == True:
                cv2.rectangle(img,(ix,iy),(x,y),(0,255,0),-1)
            else:
                cv2.circle(img,(x,y),5,(0,0,255),-1)

    #滑鼠擡起時
    elif event == cv2.EVENT_LBUTTONUP:
        drawing = False
        if mode == True:
            cv2.rectangle(img,(ix,iy),(x,y),(0,255,0),-1)
        else:
            cv2.circle(img,(x,y),5,(0,0,255),-1)


#初始化影象
img = np.zeros((512,512,3), np.uint8)
cv2.namedWindow('image')
cv2.setMouseCallback('image',draw_circle)

#按下"m"來切換模式
while(1):
    cv2.imshow('image',img)
    k = cv2.waitKey(1) & 0xFF
    if k == ord('m'):
        mode = not mode
    elif k == 27:
        break

cv2.destroyAllWindows()
  • cv2.add()——將兩張圖片的畫素疊加起來,與兩張圖片直接相加結果不同
>>> x = np.uint8([250])
>>> y = np.uint8([10])

>>> print cv2.add(x,y) # 250+10 = 260 => 255 最高為255
[[255]]

>>> print x+y          # 250+10 = 260 % 256 = 4
[4]
  • cv2.addWeighted()

未完待續