From a09fe65ed2a8b1906d31360ce9982de44ce3e600 Mon Sep 17 00:00:00 2001 From: Isaac Riley Date: Tue, 22 Feb 2022 16:04:11 +0100 Subject: [PATCH] minor changes to demo script --- vidocp/utils/deskew.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/vidocp/utils/deskew.py b/vidocp/utils/deskew.py index 3ed3f29..5d1b7b0 100644 --- a/vidocp/utils/deskew.py +++ b/vidocp/utils/deskew.py @@ -45,7 +45,7 @@ def rotate_straight(im: np.array, skew_angle: int) -> np.array: def deskew_linebased(image: np.array, verbose=False) -> np.array: skew_angle = detect_angle_from_lines(image) if verbose: - print(f"Skew angle: {skew_angle}") + print(f"Skew angle from lines: {skew_angle}") if skew_angle: deskewed = rotate_straight(image, skew_angle) return deskewed @@ -70,7 +70,7 @@ def deskew_histbased(page: np.array, max_abs_angle=1.5, delta=0.15, mode="neares best_angle = angles[scores.index(max(scores))] if verbose: - print("Best angle: {}".format(best_angle)) + print("Skew angle from pixel histogram: {}".format(best_angle)) rotated = rotate(page, best_angle, reshape=False, order=0, mode=mode) return rotated