Python opencv模块cv2安装和部分函数使用

Python opencv模块cv2安装和部分函数使用

前几天做了一下验证码识别，在这里分享一下用到的opencv模块cv2部分函数的使用方法，也是给自己加深一下记忆。

一、cv2模块安装

在这里提醒一下这里有坑欧
你如果直接用 pip install cv2 会报错欧

往下看解决办法
可以通过 pip install opencv-python 来进行安装

如果pip不能安装还可以通过 https://pypi.tuna.tsinghua.edu.cn/simple/opencv-python寻找自己python对应的 .whl文件下载进行安装，我这安装的是opencv_python-3.1.0.5-cp36-cp36m-win_amd64.whl 其中cp36是我的python版本是3.6版的，自我感觉3.6版本的.whl文件还比较好找，只是自我观点amd64是我安装的python是64位的

二、cv2模块的使用和函数介绍

1、cv2模块的使用

检测cv2是否安装成功

import cv2

如果不报错就是已经安装成功

2、cv2模块的函数介绍

下面来介绍一下cv2模块的函数介绍

（1）cv2.imread() 读入图片

参数1：图片所在位置
提示：如果想显示网页上的图片还可以写参数1还可以写成网页的网址欧

import cv2

# 读入图像
img = cv2.imread("./2.jpg")

# 显示图像
cv2.imshow("bug", img)
cv2.waitKey(10)  # 单位毫秒
cv2.destroyWindow("bug")

# 复制图像
new_img = img.copy()

# 保存图像
cv2.imwrite("bug-new.png", new_img)

（2）cv2.VideoCapture() 读取图片

参数1：可以为0和1，也可以去获取网络摄像头的网址
cv2.VideoCapture(0)表示获取电脑的摄像头
cv2.VideoCapture(1) 表示获取电脑外部连接的摄像头
cv2.VideoCapture(http://192.168.0.1:8080/?action=snapshot) 表示获取网络摄像头的视频

import cv2

cap = cv2.VideoCapture(0)   #调整参数实现读取视频或调用摄像头
while (cap.isOpened()):
    ret, frame = cap.read()
    cv2.imshow("cap", frame)
    if (cv2.waitKey(100) & 0xff) == ord('q'):   #在一个给定的时间内(单位ms)等待用户按键触发
        break
cap.release()
cv2.destroyAllWindows()

（3）cv2.cvtColor() 颜色转换

参数1：所以转换的图片

参数2：要转换的模式 cv2.COLOR_BGR2GRAY：转换为灰度图。cv2.COLOR_BGR2HSV：转换为HSV颜色空间。

（4）cv2.threshold() 二值化

参数1：要灰度的图片

参数2：阈值

参数3：最大值

参数4：转换方式 cv2.THRESH_BINARY、cv2.THRESH_BINARY_INV、cv2.THRESH_TRUNC、cv2.THRESH_TOZERO、cv2.THRESH_TOZERO_INV

（5）cv2.medianBlur() 滤波

参数1：要滤波的图片

参数2：滤波尺寸大小

（6）cv2.boundingRect() 求包含轮廓的正方框

参数1：要计算的某一轮廓

（7）cv2.findContours() 提取图片轮廓

参数1：要提取轮廓的图片

参数2：提取规则。cv2.RETR_EXTERNAL：只找外轮廓，cv2.RETR_TREE：内外轮廓都找。

参数3：输出轮廓内容格式。cv2.CHAIN_APPROX_SIMPLE：输出少量轮廓点。cv2.CHAIN_APPROX_NONE：输出大量轮廓点。

输出参数1：提取轮廓后的图片

输出参数2：轮廓列表

输出参数3：层级

下面附上我的一个验证码识别的代码

# -*- coding: utf-8 -*-

import os
import cv2
import numpy as np

def split_picture(imagepath):

    # 以灰度模式读取图片
    gray = cv2.imread(imagepath, 0)

    # 将图片的边缘变为白色
    height, width = gray.shape
    for i in range(width):
        gray[0, i] = 255
        gray[height-1, i] = 255
    for j in range(height):
        gray[j, 0] = 255
        gray[j, width-1] = 255

    # 中值滤波
    blur = cv2.medianBlur(gray, 3) #模板大小3*3

    # 二值化
    ret,thresh1 = cv2.threshold(blur, 200, 255, cv2.THRESH_BINARY)

    # 提取单个字符
    chars_list = []
    image, contours, hierarchy = cv2.findContours(thresh1, 2, 2)
    for cnt in contours:
        # 最小的外接矩形
        x, y, w, h = cv2.boundingRect(cnt)
        if x != 0 and y != 0 and w*h >= 100:
            chars_list.append((x,y,w,h))

    sorted_chars_list = sorted(chars_list, key=lambda x:x[0])
    for i,item in enumerate(sorted_chars_list):
        x, y, w, h = item
        cv2.imwrite('test_verifycode/%d.jpg'%(i+1), thresh1[y:y+h, x:x+w])

def remove_edge_picture(imagepath):

    image = cv2.imread(imagepath, 0)
    height, width = image.shape
    corner_list = [image[0,0] < 127,
                   image[height-1, 0] < 127,
                   image[0, width-1]<127,
                   image[ height-1, width-1] < 127
                   ]
    if sum(corner_list) >= 3:
        os.remove(imagepath)

def resplit_with_parts(imagepath, parts):
    image = cv2.imread(imagepath, 0)
    os.remove(imagepath)
    height, width = image.shape

    file_name = imagepath.split('/')[-1].split(r'.')[0]
    # 将图片重新分裂成parts部分
    step = width//parts     # 步长
    start = 0             # 起始位置
    for i in range(parts):
        cv2.imwrite('./test_verifycode/%s.jpg'%(file_name+'-'+str(i)), \
                    image[:, start:start+step])
        start += step

def resplit(imagepath):

    image = cv2.imread(imagepath, 0)
    height, width = image.shape

    if width >= 64:
        resplit_with_parts(imagepath, 4)
    elif width >= 48:
        resplit_with_parts(imagepath, 3)
    elif width >= 26:
        resplit_with_parts(imagepath, 2)

# rename and convert to 16*20 size
def convert(dir, file):

    imagepath = dir+'/'+file
    # 读取图片
    image = cv2.imread(imagepath, 0)
    # 二值化
    ret, thresh = cv2.threshold(image, 127, 255, cv2.THRESH_BINARY)
    img = cv2.resize(thresh, (16, 20), interpolation=cv2.INTER_AREA)
    # 保存图片
    cv2.imwrite('%s/%s' % (dir, file), img)

# 读取图片的数据，并转化为0-1值
def Read_Data(dir, file):

    imagepath = dir+'/'+file
    # 读取图片
    image = cv2.imread(imagepath, 0)
    # 二值化
    ret, thresh = cv2.threshold(image, 127, 255, cv2.THRESH_BINARY)
    # 显示图片
    bin_values = [1 if pixel==255 else 0 for pixel in thresh.ravel()]

    return bin_values

def predict(VerifyCodePath):

    dir = './test_verifycode'
    files = os.listdir(dir)

    # 清空原有的文件
    if files:
        for file in files:
            os.remove(dir + '/' + file)

    split_picture(VerifyCodePath)

    files = os.listdir(dir)
    if not files:
        print('查看的文件夹为空！')
    else:

        # 去除噪声图片
        for file in files:
            remove_edge_picture(dir + '/' + file)

        # 对黏连图片进行重分割
        for file in os.listdir(dir):
            resplit(dir + '/' + file)

        # 将图片统一调整至16*20大小
        for file in os.listdir(dir):
            convert(dir, file)

        # 图片中的字符代表的向量
        files = sorted(os.listdir(dir), key=lambda x: x[0])
        table = np.array([Read_Data(dir, file) for file in files]).reshape(-1,20,16,1)

        # 模型保存地址
        mp = './verifycode_Keras.h5'
        # 载入模型
        from keras.models import load_model
        cnn = load_model(mp)
        # 模型预测
        y_pred = cnn.predict(table)
        predictions = np.argmax(y_pred, axis=1)

        # 标签字典
        keys = range(31)
        vals = ['1', '2', '3', '4', '5', '6', '7', '8', '9', 'A', 'B', 'C', 'D', 'E', 'F', 'G', 'H', 'J', 'K', 'L', 'N',
                'P', 'Q', 'R', 'S', 'T', 'U', 'V', 'X', 'Y', 'Z']
        label_dict = dict(zip(keys, vals))

        return ''.join([label_dict[pred] for pred in predictions])

def main():

    dir = './VerifyCode/'
    correct = 0
    for i, file in enumerate(os.listdir(dir)):
        true_label = file.split('.')[0]
        VerifyCodePath = dir+file
        pred = predict(VerifyCodePath)

        if true_label == pred:
            correct += 1
        print(i+1, (true_label, pred), true_label == pred, correct)

    total = len(os.listdir(dir))
    print('\n总共图片：%d张\n识别正确：%d张\n识别准确率:%.2f%%.'\
          %(total, correct, correct*100/total))

main()

如需要下载以上代码和验证码可点击下面链接进行下载链接: https://pan.baidu.com/s/18yvxdn3_sD1tIjUrf56o9Q 提取码：ji4y
如果发现错误可联系我欧