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