qurator-spk/eynollah
1
0
Fork 0
mirror of https://github.com/qurator-spk/eynollah.git synced 2025-06-09 04:09:54 +02:00
Code Issues Projects Releases Wiki Activity

avoiding artifact images on the boundary of documents

Browse source
This commit is contained in:
vahidrezanezhad 2023-11-27 22:12:50 +01:00
parent e7d12d3549
commit 6aac0b8faf
1 changed files with 32 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

34
qurator/eynollah/eynollah.py
View file

@ -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