mirror of
https://github.com/qurator-spk/eynollah.git
synced 2025-06-08 19:59:56 +02:00
avoiding artifact images on the boundary of documents
This commit is contained in:
parent
e7d12d3549
commit
6aac0b8faf
1 changed files with 32 additions and 2 deletions
|
@ -1669,9 +1669,39 @@ class Eynollah:
|
||||||
|
|
||||||
text_regions_p_true = cv2.fillPoly(text_regions_p_true, pts = polygons_of_only_texts, color=(1,1,1))
|
text_regions_p_true = cv2.fillPoly(text_regions_p_true, pts = polygons_of_only_texts, color=(1,1,1))
|
||||||
|
|
||||||
polygons_of_images = return_contours_of_interested_region(text_regions_p_true, 2)
|
polygons_of_images = return_contours_of_interested_region(text_regions_p_true, 2, 0.0001)
|
||||||
|
|
||||||
return text_regions_p_true, erosion_hurts, polygons_lines_xml, polygons_of_images
|
image_boundary_of_doc = np.zeros((text_regions_p_true.shape[0], text_regions_p_true.shape[1]))
|
||||||
|
|
||||||
|
image_boundary_of_doc[:20, :] = 1
|
||||||
|
image_boundary_of_doc[text_regions_p_true.shape[0]-20:text_regions_p_true.shape[0], :] = 1
|
||||||
|
|
||||||
|
image_boundary_of_doc[:, :20] = 1
|
||||||
|
image_boundary_of_doc[:, text_regions_p_true.shape[1]-20:text_regions_p_true.shape[1]] = 1
|
||||||
|
|
||||||
|
#plt.imshow(image_boundary_of_doc)
|
||||||
|
#plt.show()
|
||||||
|
|
||||||
|
polygons_of_images_fin = []
|
||||||
|
for ploy_img_ind in polygons_of_images:
|
||||||
|
test_poly_image = np.zeros((text_regions_p_true.shape[0], text_regions_p_true.shape[1]))
|
||||||
|
test_poly_image = cv2.fillPoly(test_poly_image, pts = [ploy_img_ind], color=(1,1,1))
|
||||||
|
|
||||||
|
test_poly_image = test_poly_image[:,:] + image_boundary_of_doc[:,:]
|
||||||
|
test_poly_image_intersected_area = ( test_poly_image[:,:]==2 )*1
|
||||||
|
|
||||||
|
test_poly_image_intersected_area = test_poly_image_intersected_area.sum()
|
||||||
|
|
||||||
|
if test_poly_image_intersected_area==0:
|
||||||
|
polygons_of_images_fin.append(ploy_img_ind)
|
||||||
|
#plt.imshow(test_poly_image)
|
||||||
|
#plt.show()
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
return text_regions_p_true, erosion_hurts, polygons_lines_xml, polygons_of_images_fin
|
||||||
def get_regions_light_v(self,img,is_image_enhanced, num_col_classifier):
|
def get_regions_light_v(self,img,is_image_enhanced, num_col_classifier):
|
||||||
self.logger.debug("enter get_regions_light_v")
|
self.logger.debug("enter get_regions_light_v")
|
||||||
erosion_hurts = False
|
erosion_hurts = False
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue