Следующий код может помочь вам обнаружить / сегментировать страницу в изображении ...
import cv2
import matplotlib.pyplot as plt
import numpy as np
image = cv2.imread('test_p.jpg')
image = cv2.imread('test_p.jpg')
print(image.shape)
ori = image.copy()
image = cv2.resize(image, (image.shape[1]//10,image.shape[0]//10))
Изменить размер изображения, чтобы ускорить выполнение операций, чтобы мы могли работать в режиме реального времени ..
gray = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
gray = cv2.GaussianBlur(gray, (11,11), 0)
edged = cv2.Canny(gray, 75, 200)
print("STEP 1: Edge Detection")
plt.imshow(edged)
plt.show()
cnts = cv2.findContours(edged.copy(), cv2.RETR_LIST, cv2.CHAIN_APPROX_SIMPLE)
cnts = sorted(cnts[1], key = cv2.contourArea, reverse = True)[:5]
Здесь мы рассмотрим только первые 5 контуров из отсортированного списка на основе области. Здесь размер размытия по Гауссу является бит-чувствительным, поэтому выбирайте его соответствующим образом в зависимости от размера изображения.После описанных выше операций изображение может выглядеть следующим образом: *
for c in cnts:
### Approximating the contour
#Calculates a contour perimeter or a curve length
peri = cv2.arcLength(c, True)
approx = cv2.approxPolyDP(c, 0.01 * peri, True)
# if our approximated contour has four points, then we
# can assume that we have found our screen
screenCnt = approx
if len(approx) == 4:
screenCnt = approx
break
# show the contour (outline)
print("STEP 2: Finding Boundary")
cv2.drawContours(image, [screenCnt], -1, (0, 255, 0), 2)
image_e = cv2.resize(image,(image.shape[1],image.shape[0]))
cv2.imwrite('image_edge.jpg',image_e)
plt.imshow(image_e)
plt.show()
Окончательное изображение может выглядеть следующим образом ...
Остальная частьвсе может быть обработано после получения окончательного изображения ...
Код ссылки: - Git Repository
Полагаю, этот ответ будет полезен ...