diff --git a/box_detection/box_detection.py b/box_detection/box_detection.py index e1192a1..27abc76 100644 --- a/box_detection/box_detection.py +++ b/box_detection/box_detection.py @@ -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) -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) 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 = 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