Spaces:

Soumen
/

Text-Summarization-and-NLP-tasks

Running

Update line_cor.py

fdefb94 over 1 year ago

No virus

1.37 kB


	# use this command to install open cv2
	# pip install opencv-python

	# use this command to install PIL
	# pip install Pillow

	import cv2
	from PIL import Image

	def mark_region(im):

	#im = cv2.imread(image_path)

	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)

	# Dilate to combine adjacent text contours
	kernel = cv2.getStructuringElement(cv2.MORPH_RECT, (9,9))
	dilate = cv2.dilate(thresh, kernel, iterations=4)

	# Find contours, highlight text areas, and extract ROIs
	cnts = cv2.findContours(dilate, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)
	cnts = cnts[0] if len(cnts) == 2 else cnts[1]

	line_items_coordinates = []
	image=None
	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