diff --git a/box_detection/redaction_detection.py b/box_detection/redaction_detection.py index 27abc76..6b1c390 100644 --- a/box_detection/redaction_detection.py +++ b/box_detection/redaction_detection.py @@ -54,8 +54,8 @@ def annotate_boxes_in_pdf(pdf_path, page_index=1): page = pdf2image.convert_from_path(pdf_path, first_page=page_index + 1, last_page=page_index + 1)[0] page = np.array(page) - asd = find_redactions(page) - page = annotate_poly(page, asd) + redaction_contours = find_redactions(page) + page = annotate_poly(page, redaction_contours) fig, ax = plt.subplots(1, 1) fig.set_size_inches(20, 20)