python – 无法使用sort_contors构建七段OCR

我正在尝试构建一个用于识别七段显示的OCR,如下所述

Original Image

使用开放式CV的预处理工具我在这里得到它

threshold

现在我正在尝试按照本教程 – https://www.pyimagesearch.com/2017/02/13/recognizing-digits-with-opencv-and-python/

但就此而言

digitCnts = contours.sort_contours(digitCnts,method="left-to-right")[0]
digits = []

我收到错误

使用THRESH_BINARY_INV解决错误,但仍然没有OCR工作任何修复都会很好

在sort_contours中输入文件“/Users/ms/anaconda3/lib/python3.6/site-packages/imutils/contours.py”,第25行
    key = lambda b:b1 [i],reverse = reverse))

ValueError:没有足够的值来解包(预期2,得到0)

任何想法如何解决这个问题,让我的OCR成为一个有效的模型

我的整个代码是:

import numpy as np 
import cv2
import imutils
# import the necessary packages
from imutils.perspective import four_point_transform
from imutils import contours
import imutils
import cv2

# define the dictionary of digit segments so we can identify
# each digit on the thermostat
DIGITS_LOOKUP = {
    (1,1,1): 0,(0,0): 1,(1,0): 2,1): 3,0): 4,1): 5,1): 6,0): 7,1): 8,1): 9
}

# load image
image = cv2.imread('d4.jpg')
# create hsv
hsv = cv2.cvtColor(image,cv2.COLOR_BGR2HSV)

 # set lower and upper color limits
low_val = (60,180,160)
high_val = (179,255,255)
# Threshold the HSV image 
mask = cv2.inRange(hsv,low_val,high_val)
# find contours in mask
ret,cont,hierarchy = cv2.findContours(mask,cv2.RETR_EXTERNAL,cv2.CHAIN_APPROX_SIMPLE)

# select the largest contour
largest_area = 0
for cnt in cont:
    if cv2.contourArea(cnt) > largest_area:
        cont = cnt
        largest_area = cv2.contourArea(cnt)

# get the parameters of the boundingBox
x,y,w,h = cv2.boundingRect(cont)

# create and show subimage
roi = image[y:y+h,x:x+w]
cv2.imshow("Result",roi)


#  draw Box on original image and show image
cv2.rectangle(image,(x,y),(x+w,y+h),255),2)
cv2.imshow("Image",image)

grayscaled = cv2.cvtColor(roi,cv2.COLOR_BGR2GRAY)
retval,threshold = cv2.threshold(grayscaled,10,cv2.THRESH_BINARY)
retval2,threshold2 = cv2.threshold(grayscaled,125,cv2.THRESH_BINARY+cv2.THRESH_OTSU)
cv2.imshow('threshold',threshold2)
cv2.waitKey(0)
cv2.destroyAllWindows()
# find contours in the thresholded image,then initialize the
# digit contours lists
cnts = cv2.findContours(threshold2.copy(),cv2.CHAIN_APPROX_SIMPLE)
cnts = imutils.grab_contours(cnts)
digitCnts = []

# loop over the digit area candidates
for c in cnts:
    # compute the bounding Box of the contour
    (x,h) = cv2.boundingRect(c)
    # if the contour is sufficiently large,it must be a digit
    if w >= 15 and (h >= 30 and h <= 40):
        digitCnts.append(c)
# sort the contours from left-to-right,then initialize the
# actual digits themselves
digitCnts = contours.sort_contours(digitCnts,method="left-to-right")[0]
digits = []


# loop over each of the digits
for c in digitCnts:
    # extract the digit ROI
    (x,h) = cv2.boundingRect(c)
    roi = thresh[y:y + h,x:x + w]

    # compute the width and height of each of the 7 segments
    # we are going to examine
    (roiH,roiW) = roi.shape
    (dW,dH) = (int(roiW * 0.25),int(roiH * 0.15))
    dHC = int(roiH * 0.05)

    # define the set of 7 segments
    segments = [
        ((0,0),(w,dH)),# top
        ((0,(dW,h // 2)),# top-left
        ((w - dW,# top-right
        ((0,(h // 2) - dHC),(h // 2) + dHC)),# center
        ((0,h // 2),h)),# bottom-left
        ((w - dW,# bottom-right
        ((0,h - dH),h))   # bottom
    ]
    on = [0] * len(segments)

    # loop over the segments
    for (i,((xA,yA),(xB,yB))) in enumerate(segments):
        # extract the segment ROI,count the total number of
        # thresholded pixels in the segment,and then compute
        # the area of the segment
        segROI = roi[yA:yB,xA:xB]
        total = cv2.countNonZero(segROI)
        area = (xB - xA) * (yB - yA)

        # if the total number of non-zero pixels is greater than
        # 50% of the area,mark the segment as "on"
        if total / float(area) > 0.5:
            on[i]= 1

    # lookup the digit and draw it on the image
    digit = DIGITS_LOOKUP[tuple(on)]
    digits.append(digit)
    cv2.rectangle(output,(x + w,y + h),1)
    cv2.putText(output,str(digit),(x - 10,y - 10),cv2.FONT_HERShey_SIMPLEX,0.65,2)
# display the digits
print(u"{}{}.{}{}.{}{} \u00b0C".format(*digits))
cv2.imshow("Input",image)
cv2.imshow("Output",output)
cv2.waitKey(0)

帮助将很好地修复我的OCR

最佳答案
所以,正如我在评论中所说,有两个问题:

>您试图在白色背景上找到黑色轮廓,与OpenCV documentation相反.这是使用THRESH_BINARY_INV标志而不是THRESH_BINARY解决的.
>由于未连接数字,因此无法找到该数字的完整轮廓.所以我尝试了一些形态学操作.以下是步骤:

First Threshold

2a)使用以下代码打开上面的图像:

threshold2 = cv2.morphologyEx(threshold,cv2.MORPH_OPEN,np.ones((3,3),np.uint8))

Opening

2b)上一张图片的扩张:

threshold2 = cv2.dilate(threshold2,np.ones((5,1),np.uint8),iterations=1)

Dilation

2c)由于扩展到顶部边框,裁剪图像的顶部以分隔数字:

height,width = threshold2.shape[:2]
threshold2 = threshold2[5:height,5:width]

注意不知何故,图像显示在这里没有我正在谈论的白色边框.尝试在新窗口中打开图像,您将看到我的意思.

Final cropping

因此,在解决了这些问题之后,轮廓非常好,它们应该如何在这里看到:

cnts = cv2.findContours(threshold2.copy(),cv2.CHAIN_APPROX_SIMPLE)
cnts = imutils.grab_contours(cnts)

digitCnts = []

# loop over the digit area candidates
for c in cnts:
    # compute the bounding Box of the contour
    (x,it must be a digit
    if w <= width * 0.5 and (h >= height * 0.2):
        digitCnts.append(c)
# sort the contours from left-to-right,then initialize the
# actual digits themselves
cv2.drawContours(image2,digitCnts,-1,255))
cv2.imwrite("cnts-sort.jpg",image2)

如下所示,轮廓以红色绘制.

Contours

现在,为了估计数字是否是代码,这部分不知何故不起作用,我责怪查找表.从下图中可以看出,所有数字的边界值都被正确裁剪,但查找表无法识别它们.

# loop over each of the digits
j = 0
for c in digitCnts:
    # extract the digit ROI
    (x,h) = cv2.boundingRect(c)
    roi = threshold2[y:y + h,x:x + w]
    cv2.imwrite("roi" + str(j) + ".jpg",roi)
    j += 1

    # compute the width and height of each of the 7 segments
    # we are going to examine
    (roiH,mark the segment as "on"
        if area != 0:
            if total / float(area) > 0.5:
                on[i] = 1

    # lookup the digit and draw it on the image
    try:
        digit = DIGITS_LOOKUP[tuple(on)]
        digits.append(digit)
        cv2.rectangle(roi,1)
        cv2.putText(roi,2)
    except KeyError:
        continue

我通读了website you mentioned in the question,从评论看来,LUT中的一些条目可能是错误的.所以我要留给你解决这个问题.以下是找到的个别数字(但未被识别):

1

7

5

8

5

1

1

或者,您可以使用tesseract来识别这些检测到的数字.

希望能帮助到你!

版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。

相关推荐


我最近重新拾起了计算机视觉,借助Python的opencv还有face_recognition库写了个简单的图像识别demo,额外定制了一些内容,原本想打包成exe然后发给朋友,不过在这当中遇到了许多小问题,都解决了,记录一下踩过的坑。 1、Pyinstaller打包过程当中出现warning,跟d
说到Pooling,相信学习过CNN的朋友们都不会感到陌生。Pooling在中文当中的意思是“池化”,在神经网络当中非常常见,通常用的比较多的一种是Max Pooling,具体操作如下图: 结合图像理解,相信你也会大概明白其中的本意。不过Pooling并不是只可以选取2x2的窗口大小,即便是3x3,
记得大一学Python的时候,有一个题目是判断一个数是否是复数。当时觉得比较复杂不好写,就琢磨了一个偷懒的好办法,用异常处理的手段便可以大大程度帮助你简短代码(偷懒)。以下是判断整数和复数的两段小代码: 相信看到这里,你也有所顿悟,能拓展出更多有意思的方法~
文章目录 3 直方图Histogramplot1. 基本直方图的绘制 Basic histogram2. 数据分布与密度信息显示 Control rug and density on seaborn histogram3. 带箱形图的直方图 Histogram with a boxplot on t
文章目录 5 小提琴图Violinplot1. 基础小提琴图绘制 Basic violinplot2. 小提琴图样式自定义 Custom seaborn violinplot3. 小提琴图颜色自定义 Control color of seaborn violinplot4. 分组小提琴图 Group
文章目录 4 核密度图Densityplot1. 基础核密度图绘制 Basic density plot2. 核密度图的区间控制 Control bandwidth of density plot3. 多个变量的核密度图绘制 Density plot of several variables4. 边
首先 import tensorflow as tf tf.argmax(tenso,n)函数会返回tensor中参数指定的维度中的最大值的索引或者向量。当tensor为矩阵返回向量,tensor为向量返回索引号。其中n表示具体参数的维度。 以实际例子为说明: import tensorflow a
seaborn学习笔记章节 seaborn是一个基于matplotlib的Python数据可视化库。seaborn是matplotlib的高级封装,可以绘制有吸引力且信息丰富的统计图形。相对于matplotlib,seaborn语法更简洁,两者关系类似于numpy和pandas之间的关系,seabo
Python ConfigParser教程显示了如何使用ConfigParser在Python中使用配置文件。 文章目录 1 介绍1.1 Python ConfigParser读取文件1.2 Python ConfigParser中的节1.3 Python ConfigParser从字符串中读取数据
1. 处理Excel 电子表格笔记(第12章)(代码下载) 本文主要介绍openpyxl 的2.5.12版处理excel电子表格,原书是2.1.4 版,OpenPyXL 团队会经常发布新版本。不过不用担心,新版本应该在相当长的时间内向后兼容。如果你有新版本,想看看它提供了什么新功能,可以查看Open
1. 发送电子邮件和短信笔记(第16章)(代码下载) 1.1 发送电子邮件 简单邮件传输协议(SMTP)是用于发送电子邮件的协议。SMTP 规定电子邮件应该如何格式化、加密、在邮件服务器之间传递,以及在你点击发送后,计算机要处理的所有其他细节。。但是,你并不需要知道这些技术细节,因为Python 的
文章目录 12 绘图实例(4) Drawing example(4)1. Scatterplot with varying point sizes and hues(relplot)2. Scatterplot with categorical variables(swarmplot)3. Scat
文章目录 10 绘图实例(2) Drawing example(2)1. Grouped violinplots with split violins(violinplot)2. Annotated heatmaps(heatmap)3. Hexbin plot with marginal dist
文章目录 9 绘图实例(1) Drawing example(1)1. Anscombe’s quartet(lmplot)2. Color palette choices(barplot)3. Different cubehelix palettes(kdeplot)4. Distribution
Python装饰器教程展示了如何在Python中使用装饰器基本功能。 文章目录 1 使用教程1.1 Python装饰器简单示例1.2 带@符号的Python装饰器1.3 用参数修饰函数1.4 Python装饰器修改数据1.5 Python多层装饰器1.6 Python装饰器计时示例 2 参考 1 使
1. 用GUI 自动化控制键盘和鼠标第18章 (代码下载) pyautogui模块可以向Windows、OS X 和Linux 发送虚拟按键和鼠标点击。根据使用的操作系统,在安装pyautogui之前,可能需要安装一些其他模块。 Windows: 不需要安装其他模块。OS X: sudo pip3
文章目录 生成文件目录结构多图合并找出文件夹中相似图像 生成文件目录结构 生成文件夹或文件的目录结构,并保存结果。可选是否滤除目录,特定文件以及可以设定最大查找文件结构深度。效果如下: root:[z:/] |--a.py |--image | |--cat1.jpg | |--cat2.jpg |
文章目录 VENN DIAGRAM(维恩图)1. 具有2个分组的基本的维恩图 Venn diagram with 2 groups2. 具有3个组的基本维恩图 Venn diagram with 3 groups3. 自定义维恩图 Custom Venn diagram4. 精致的维恩图 Elabo
mxnet60分钟入门Gluon教程代码下载,适合做过深度学习的人使用。入门教程地址: https://beta.mxnet.io/guide/getting-started/crash-course/index.html mxnet安装方法:pip install mxnet 1 在mxnet中使
文章目录 1 安装2 快速入门2.1 基本用法2.2 输出图像格式2.3 图像style设置2.4 属性2.5 子图和聚类 3 实例4 如何进一步使用python graphviz Graphviz是一款能够自动排版的流程图绘图软件。python graphviz则是graphviz的python实