Как обрезать изображение на части после определения краев с помощью Python

Question

Я работаю над проектом реконструкции разорванного документа.Сначала я попытался определить края изображения, которые содержат порванные части документа, а затем я попытался обрезать изображение на части через обнаруженные края, используя образец кода,

import cv2
import numpy as np
img = cv2.imread("test.png")
img = cv2.imread("d:/test.jpeg")

cv2.imshow('Original Image',img)

new_img = cv2.Canny(img, 0, 505)
cv2.imshow('new image', new_img)

blurred = cv2.blur(new_img, (3,3))
canny = cv2.Canny(blurred, 50, 200)

## find the non-zero min-max coords of canny
pts = np.argwhere(canny>0)
y1,x1 = pts.min(axis=0)
y2,x2 = pts.max(axis=0)

## crop the region
cropped = new_img[y1:y2, x1:x2]
cv2.imwrite("cropped.png", cropped)

tagged = cv2.rectangle(new_img.copy(), (x1,y1), (x2,y2), (0,255,0), 3, cv2.LINE_AA)
cv2.imshow("tagged", tagged)
cv2.waitKey()

мое входное изображение было

после выполнения вышеуказанного кода я получаю вывод наподобие

Может кто-нибудь помочь мне обрезать порванные куски документа и назначитьих в переменные

Answer 1

Начало моего рабочего процесса похоже на ваше. Первый шаг: размытие изображения ..

blurred = cv2.GaussianBlur(gray, (5, 5), 0) # Blur

Второй шаг: получить хитрый образ ...

canny = cv2.Canny(blurred, 30, 150) # Canny

Третий шаг: нарисуйте контуры на хитрый образ. Это закрывает разорванные кусочки.

# Find contours
_, contours, _ = cv2.findContours(canny,cv2.RETR_TREE,cv2.CHAIN_APPROX_SIMPLE)
# Draw contours on canny (this connects the contours
cv2.drawContours(canny, contours, -1, 255, 2)
canny = 255 - canny

Четвертый шаг: заливка (заполненные участки серые)

# Get mask for floodfill
h, w = thresh.shape[:2]
mask = np.zeros((h+2, w+2), np.uint8)
# Floodfill from point (0, 0)
cv2.floodFill(thresh, mask, (0,0), 123);

Пятый шаг: избавиться от действительно маленьких и действительно больших контуров

# Create a blank image to draw on
res = np.zeros_like(src_img)
# Create a list for unconnected contours
unconnectedContours = []
for contour in contours:
    area = cv2.contourArea(contour)
    # If the contour is not really small, or really big
    if area > 123 and area < 760000:
        cv2.drawContours(res, [contour], 0, (255,255,255), cv2.FILLED)
        unconnectedContours.append(contour)

Наконец, после того, как вы разбили фрагменты, они могут быть вложены.