Opencv и python получить правильный контур документа и нарисовать изображение - PullRequest
Новая
       79

Opencv и python получить правильный контур документа и нарисовать изображение

0 голосов
/ 08 мая 2020

Используя opencv с python, вот результат документа conture enter image description here

Я использую opencv_python-4.2.0.34, нижний левый угол не резкий, как другие углы. Как это исправить? (отредактируйте левый нижний угол автоматически, чтобы он выглядел резким, по мере появления других)

вот код для рисования контура: 

img = cv2.imread(imagePath)
gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
invGamma = 1.0 / 0.3
table = np.array([((i / 255.0) ** invGamma) * 255
                  for i in np.arange(0, 256)]).astype("uint8")

# apply gamma correction using the lookup table
gray = cv2.LUT(gray, table)

ret, thresh1 = cv2.threshold(gray, 80, 255, cv2.THRESH_BINARY)

contours, hierachy = cv2.findContours(thresh1, cv2.RETR_TREE, cv2.CHAIN_APPROX_SIMPLE)

# except:
#     print('exception occurred')

def biggestRectangle(conts):
    biggest = None
    max_area = 0
    indexReturn = -1
    for index in range(len(conts)):
        i = conts[index]
        area = cv2.contourArea(i)
        if area > 100:
            peri = cv2.arcLength(i, True)
            approx = cv2.approxPolyDP(i, 0.1 * peri, True)
            if area > max_area and len(approx) == 4:
                biggest = approx
                max_area = area
                indexReturn = index
    return indexReturn

indexReturn = biggestRectangle(contours)
hull = cv2.convexHull(contours[indexReturn])

print(indexReturn)
print(contours[indexReturn])
print(hull)

# copyImg = img.copy()

cv2.imwrite(os.path.join('path_to_save', imageName),
            cv2.drawContours(img, [hull], 0, (0, 255, 0), 3))

Как сделать его правильным. Я пробовал использовать другой подход: 

x, y, w, h = cv2.boundingRect(hull)

    cv2.rectangle(img, (x, y), (x+w, y+h), (255, 0, 0), 2)
    newImageName = imageName.split('.')
    newImageName = newImageName[0] + '_rect.' + newImageName[1]
    print(newImageName)
    cv2.imwrite(os.path.join('path_to_save', newImageName), img)

, но он дает мне это (синий контур): enter image description here

Что тоже неверно. Как это исправить?

1 Ответ

3 голосов
/ 08 мая 2020

Поскольку вы знаете, что интересующий вас контур - это лист бумаги, возможно, вы можете использовать более систематизированный c подход, чтобы найти самый большой повернутый прямоугольник с помощью OpenCV cv2.minAreaRect ( ссылка ) . Он вернет центр, размер и поворот прямоугольника, наиболее плотно охватывающего ваш контур. Поскольку у вашего изображения довольно сложный фон, мне пришлось применить еще несколько шагов обработки, используя функцию OpenCV morphologyEx , прежде чем захватывать контуры, но в этом коде я мог увидеть лучший результат (показано ниже). 

# first, do some processing
kernel1 = cv2.getStructuringElement(cv2.MORPH_ELLIPSE, (27, 27))
kernel2 = cv2.getStructuringElement(cv2.MORPH_ELLIPSE, (13, 13))
thresh1 = cv2.morphologyEx(thresh1, cv2.MORPH_ERODE, kernel1)
thresh1 = cv2.morphologyEx(thresh1, cv2.MORPH_DILATE, kernel2)

# find all contours
contours, hierarchy = cv2.findContours(thresh1, cv2.RETR_LIST, 
                                       cv2.CHAIN_APPROX_SIMPLE)
# Find the largest ('max' using the key of the contour area)
biggest_contour = max(contours, key=cv2.contourArea)

# Calculate the minimum enclosing rectangle: format ((x, y), (length, width), angle)
rectangle = cv2.minAreaRect(biggest_contour)

# conver the minAreaRect output to points
pts = cv2.boxPoints(rectangle)
# contours must be of shape (n, 1, 2) and dtype integer
pts = pts.reshape(-1, 1, 2)
pts = pts.astype(int)

# draw contours and display
cv2.drawContours(img, [pts], -1, (255, 0, 0), 2)
cv2.imshow('img', img)
k = cv2.waitKey(0)

вывод:
enter image description here

...