本文主要是介绍【opencv】信用卡号识别实验,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!
实验环境:anaconda、jupyter notebook(其它的ide也行)
实验用的包:numpy、matplotlib、opencv
实验目标:
识别信用卡的卡号
信用卡图片:
数字模板图片:
一、包引入
import cv2
import matplotlib.pyplot as plt
import numpy as np
二、数字模板特征提取
图片二值化处理
template = cv2.imread('template.png')# 灰度处理
template_gray = cv2.cvtColor(template, cv2.COLOR_BGR2GRAY)
# 二值处理
ret,template_bin = cv2.threshold(template_gray,127,255,cv2.THRESH_BINARY_INV)plt.imshow(template_bin, 'gray')
plt.show()
检测数字模板外轮廓
# 只检测外轮廓
binary, template_contours, hierarchy = cv2.findContours(template_bin, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_NONE)
template_copy = template.copy()template_contours_img = cv2.drawContours(template_copy, template_contours, -1, (0,0,255),1)plt.imshow(template_contours_img)
plt.show()
找到外接矩形
# 外接矩形
tangles = []
template_copy = template.copy()
for cnt in template_contours:x,y,w,h = cv2.boundingRect(cnt)tangles.append((x,y,w,h))template_copy = cv2.rectangle(template_copy, (x, y), (x + w, y + h), (0,255,0), 2)
plt.imshow(cv2.cvtColor(template_copy, cv2.COLOR_BGR2RGB))
plt.show()
# 根据x值狠狠排序
tangles.sort()
print(tangles)
通过外接矩形截取数字图片
# 拿到数字对应的图片
number_size = (50,100)
digits = {}
for i in range(len(tangles)):(x,y,w,h) = tangles[i]digits[i] = cv2.resize(template_bin[y:y+h, x: x+w], number_size)plt.subplot(1,10,1 + i)plt.xticks([])plt.yticks([])plt.imshow(digits[i],'gray')
plt.show()
三、信用卡卡面特征提取
初始化卷积核、处理卡片
# 初始化卷积核
rectKernel = cv2.getStructuringElement(cv2.MORPH_RECT,(9,3))
sqKernel = cv2.getStructuringElement(cv2.MORPH_RECT,(5,5))# 读入信用卡图片
card = cv2.imread('card.png')
# 处理为灰度图
card_gray = cv2.cvtColor(card, cv2.COLOR_BGR2GRAY)
# 处理为礼帽处理
card_tophat = cv2.morphologyEx(card_gray, cv2.MORPH_TOPHAT, rectKernel)plt.imshow(card_tophat,'gray')
plt.show()
对信用卡图片梯度处理
gradx = cv2.Sobel(card_tophat, ddepth=cv2.CV_32F, dx=1,dy=0, ksize=-1)
card_gradx = np.absolute(gradx)
(min_val,max_val) = (np.min(card_gradx), np.max(card_gradx))
card_gradx = (255 * ((card_gradx - min_val) / (max_val - min_val)))
card_gradx = card_gradx.astype('uint8')plt.imshow(card_gradx,'gray')
plt.show()
把特征轮廓合成一片,方便提取
闭操作
card_closing = cv2.morphologyEx(card_gradx, cv2.MORPH_CLOSE,rectKernel)plt.imshow(card_closing,'gray')
plt.show()
二值化
# 二值化,自动判断
ret,card_bin = cv2.threshold(card_closing,0,255,cv2.THRESH_BINARY | cv2.THRESH_OTSU)plt.imshow(card_bin,'gray')
plt.show()
膨胀操作
# 膨胀操作,加强卡号特征
card_dilate = cv2.dilate(card_bin, rectKernel, iterations=1)plt.imshow(card_dilate,'gray')
plt.show()
获取外轮廓
# 检测外轮廓
binary, card_contours, hierarchy = cv2.findContours(card_dilate, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_NONE)
card_copy = card.copy()card_contours_img = cv2.drawContours(card_copy, card_contours, -1, (0,0,255),2)plt.imshow(cv2.cvtColor(card_contours_img,cv2.COLOR_BGR2RGB))
plt.show()
此时拿到的外轮廓有很多不需要的地方
获取需要的外轮廓
card_copy = card.copy()
locs = []
for (i, sit) in enumerate(card_contours):(x,y,w,h) = cv2.boundingRect(sit)ar = w / float(h)if ar > 2.5 and ar < 4.0:if 65 < w < 70 and 15 < h < 25:locs.append((x,y,w,h))
locs.sort()
for (x,y,w,h) in locs:card_copy = cv2.rectangle(card_copy, (x, y), (x + w, y + h), (0,255,0), 2)plt.imshow(cv2.cvtColor(card_copy, cv2.COLOR_BGR2RGB))
plt.show()
四、卡号识别
output = []
card_copy = card.copy()
for (i,(x,y,w,h)) in enumerate(locs):group_output = []# 获取每组卡号group = card_gray[y - 5 : y + h + 5, x - 5 : x + w + 5]# 转为二值图像ret, group = cv2.threshold(group,0,255,cv2.THRESH_BINARY | cv2.THRESH_OTSU)# 外轮廓检测binary, group_contours, hierarchy = cv2.findContours(group, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_NONE)data_sites = []for data_sit in group_contours:data_sites.append(cv2.boundingRect(data_sit))data_sites.sort()for (gx, gy, gw, gh) in data_sites:data = cv2.resize(group[gy : gy + gh, gx : gx + gw], number_size)scores = []# 计算分数for (digit, digit_img) in digits.items():result = cv2.matchTemplate(data, digit_img, cv2.TM_CCOEFF_NORMED)(_,score,_,_) = cv2.minMaxLoc(result)scores.append(score)group_output.append(str(np.argmax(scores)))cv2.rectangle(card_copy,(x - 5, y - 5), (x + w +5, y + h + 5), (0,255,0),2)cv2.putText(card_copy, "".join(group_output), (x, y - 15), cv2.FONT_HERSHEY_SIMPLEX, 0.65, (0,0,255),2)output.extend(group_output)plt.imshow(cv2.cvtColor(card_copy, cv2.COLOR_BGR2RGB))
plt.show()
print(output)
成功识别卡号
这篇关于【opencv】信用卡号识别实验的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!