下面是我最近在练习七段数码管识别时,在CSDN中找到的二值转化和形状识别的代码,但是在调试中形状识别一直存在问题。形状识别使用RETR_EXTERNAL参数 基本上就是只能识别一个外框, 只能改成RETR_TREE方式,但是RETR_TREE方式内部的形状也会识别出来,还要进行尺寸判断。 最终发现是不同的原图,在使用threshold函数时,要分别使用cv2.THRESH_BINARY_INV和cv2.THRESH_BINARY函数,这两个参数导致输出二值图完全相反。形状识别最好是黑底白字。 如果是白底黑字,那么使用RETR_EXTERNAL的findContours函数识别的轮廓就有问题了。
另外,部分数码管因为辉光的缘故会粘连,但是调整threshold的参数阈值不起作用,网上的资料都是腐蚀、膨胀、开、闭。但是作用不大。 这种情况下,调用threshold函数就不能带参数cv2.THRESH_OTSU,只使用cv2.THRESH_BINARY或者cv2.THRESH_BINARY_INV。 把阈值调整到200-230左右,二值化的图像就没有粘连了。
thresh = cv2.threshold(gray, 0, 255, cv2.THRESH_BINARY_INV| cv2.THRESH_OTSU)[1]
kernel = cv2.getStructuringElement(cv2.MORPH_ELLIPSE, (1, 5))
thresh = cv2.morphologyEx(thresh, cv2.MORPH_CLOSE, kernel)
# find contours in the thresholded image, then initialize the
# digit contours lists
cnts = cv2.findContours(thresh.copy(), cv2.RETR_EXTERNAL, \
cv2.CHAIN_APPROX_SIMPLE)
cnts = imutils.grab_contours(cnts)
补充说明: 我采用了imutils库,包括grab_contours和four_point_transform都是用的imutils库