This commit is contained in:
Matthias Bisping 2022-02-05 14:31:46 +01:00
parent 2aced51dfd
commit c435d25ac0

View File

@ -26,9 +26,9 @@ def is_likely_redaction(contour, hierarchy, min_area):
return is_filled(hierarchy) and is_boxy(contour) and is_large_enough(contour, min_area) return is_filled(hierarchy) and is_boxy(contour) and is_large_enough(contour, min_area)
def find_redactions(image: np.array, min_nomralized_area=200000): def find_redactions(image: np.array, min_normalized_area=200000):
min_nomralized_area /= 200 # Assumes 200 DPI PDF -> image conversion resolution min_normalized_area /= 200 # Assumes 200 DPI PDF -> image conversion resolution
gray = ~cv2.cvtColor(image, cv2.COLOR_BGR2GRAY) gray = ~cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
blurred = cv2.GaussianBlur(gray, (5, 5), 1) blurred = cv2.GaussianBlur(gray, (5, 5), 1)
@ -37,7 +37,7 @@ def find_redactions(image: np.array, min_nomralized_area=200000):
contours, hierarchies = cv2.findContours(thresh.copy(), cv2.RETR_TREE, cv2.CHAIN_APPROX_NONE) contours, hierarchies = cv2.findContours(thresh.copy(), cv2.RETR_TREE, cv2.CHAIN_APPROX_NONE)
contours = map( contours = map(
first, starfilter(partial(is_likely_redaction, min_area=min_nomralized_area), zip(contours, hierarchies[0])) first, starfilter(partial(is_likely_redaction, min_area=min_normalized_area), zip(contours, hierarchies[0]))
) )
return contours return contours