Detecting barcodes in images.
利用了图像处理中的膨胀、腐蚀运算，将条形码缺陷瑕疵，划痕缺口检测并框出来。
input
output
实现代码如下：
#!/usr/bin/env python2
# -*- coding: utf-8 -*-
"""
Created on 17-12-20 7:33 PM
barcode_defect_detection.py --检测条形码缺陷瑕疵
@author: lemoner20
"""
import numpy as np
import argparse
import cv2
# Instruction
# python barcode_defect_detection.py --image images/barcode_1.png
image = cv2.imread('./images/barcode_1.png')
gray = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
gaussianBlur = cv2.GaussianBlur(gray, (3, 3), 0)
gradX = cv2.Sobel(gaussianBlur, ddepth=cv2.CV_32F, dx=1, dy=0, ksize=-1)
gradY = cv2.Sobel(gaussianBlur, ddepth=cv2.CV_32F, dx=0, dy=1, ksize=-1)
sobelX = cv2.convertScaleAbs(gradX)
sobelY = cv2.convertScaleAbs(gradY)
grad = cv2.subtract(sobelX, sobelY)
sobel = cv2.convertScaleAbs(grad)
blurred = cv2.blur(sobel, (3, 3))
_, barcodeThresh = cv2.threshold(blurred, 180, 255, cv2.THRESH_BINARY)
barcodeKernel = cv2.getStructuringElement(cv2.MORPH_RECT, (50, 10))
closed = cv2.morphologyEx(barcodeThresh, cv2.MORPH_CLOSE, barcodeKernel)
closed = cv2.erode(closed, None, iterations=4)
closed = cv2.dilate(closed, None, iterations=4)
_, barcodeContours, _ = cv2.findContours(closed.copy(), cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)
barcodeCnt1 = sorted(barcodeContours, key=cv2.contourArea, reverse=True)[0]
barcodeCnt2 = sorted(barcodeContours, key=cv2.contourArea, reverse=True)[1]
center1, size1, angle1 = cv2.minAreaRect(barcodeCnt1)
center2, size2, angle2 = cv2.minAreaRect(barcodeCnt2)
# rect1 = cv2.minAreaRect(barcodeCnt1)
# box1 = np.int0(cv2.boxPoints(rect1))
# cv2.drawContours(image, [box1], -1, (255, 0, 0), 2)
# rect2 = cv2.minAreaRect(barcodeCnt2)
# box2 = np.int0(cv2.boxPoints(rect2))
# cv2.drawContours(image, [box2], -1, (255, 0, 0), 2)
# cv2.imshow("minRect Image", image);
rotated = cv2.getRotationMatrix2D(center2, angle2, 1)
rows, cols, _ = image.shape
rotatedImage = cv2.warpAffine(image, rotated, (cols, rows))
if abs(angle1) > 45:
rect1 = (center1, size1, 90)
box1 = np.int0(cv2.boxPoints(rect1))
temp = [[0, 0]]*4
for i in range(0, 3):
temp[i+1] = box1[i]
temp[0] = box1[3]
box1 = np.array(temp)
cv2.drawContours(rotatedImage, [box1], -1, (255, 0, 0), 2)
else:
rect1 = (center1, size1, 0)
box1 = np.int0(cv2.boxPoints(rect1))
cv2.drawContours(rotatedImage, [box1], -1, (255, 0, 0), 2)
if abs(angle2) > 45:
rect2 = (center2, size2, 90)
box2 = np.int0(cv2.boxPoints(rect2))
temp = [[0, 0]]*4
for i in range(0, 3):
temp[i+1] = box2[i]
temp[0] = box2[3]
box2 = np.array(temp)
cv2.drawContours(rotatedImage, [box2], -1, (255, 0, 0), 2)
else:
rect2 = (center2, size2, 0)
box2 = np.int0(cv2.boxPoints(rect2))
cv2.drawContours(rotatedImage, [box2], -1, (255, 0, 0), 2)
ROI1 = rotatedImage[box1[1][1]+5:box1[0][1]-5, box1[1][0]:box1[2][0]]
ROI2 = rotatedImage[box2[1][1]+5:box2[0][1]-5, box2[1][0]:box2[2][0]]
def detect(ROI):
defectKernel = cv2.getStructuringElement(cv2.MORPH_RECT, (1, 30))
eroded = cv2.erode(ROI, defectKernel)
defect = cv2.subtract(ROI, eroded)
_, defectThresh = cv2.threshold(defect, 25, 255, cv2.THRESH_BINARY)
# cv2.imshow("defectThresh Image", defectThresh);
b, g, r = cv2.split(defectThresh)
merged = cv2.merge([r-r, r-r, r])
defectGray = cv2.cvtColor(merged, cv2.COLOR_BGR2GRAY)
_, defectGray = cv2.threshold(defectGray, 50, 255, cv2.THRESH_BINARY)
_, defectContours, _ = cv2.findContours(defectGray.copy(), cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)
for ci in defectContours:
if cv2.contourArea(ci) < 10:
continue
x, y, w, h = cv2.boundingRect(ci)
cv2.rectangle(ROI, (x, y), (x+w, y+h), (0, 0, 255), 2)
detect(ROI1)
detect(ROI2)
cv2.imshow("ROIDetected Image", rotatedImage);
cv2.waitKey(0)
cv2.destroyAllWindows()