|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
import cv2 |
|
from PIL import Image |
|
|
|
def mark_region(im): |
|
|
|
|
|
|
|
gray = cv2.cvtColor(im, cv2.COLOR_BGR2GRAY) |
|
blur = cv2.GaussianBlur(gray, (9,9), 0) |
|
thresh = cv2.adaptiveThreshold(blur,255,cv2.ADAPTIVE_THRESH_GAUSSIAN_C, cv2.THRESH_BINARY_INV,11,30) |
|
|
|
|
|
kernel = cv2.getStructuringElement(cv2.MORPH_RECT, (9,9)) |
|
dilate = cv2.dilate(thresh, kernel, iterations=4) |
|
|
|
|
|
cnts = cv2.findContours(dilate, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE) |
|
cnts = cnts[0] if len(cnts) == 2 else cnts[1] |
|
|
|
line_items_coordinates = [] |
|
for c in cnts: |
|
area = cv2.contourArea(c) |
|
x,y,w,h = cv2.boundingRect(c) |
|
|
|
if y >= 600 and x <= 1000: |
|
if area > 10000: |
|
image = cv2.rectangle(im, (x,y), (2200, y+h), color=(255,0,255), thickness=3) |
|
line_items_coordinates.append([(x,y), (2200, y+h)]) |
|
|
|
if y >= 2400 and x<= 2000: |
|
image = cv2.rectangle(im, (x,y), (2200, y+h), color=(255,0,255), thickness=3) |
|
line_items_coordinates.append([(x,y), (2200, y+h)]) |
|
|
|
|
|
return image, line_items_coordinates |