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()

未完待续