Marmelator/cv-analysis-service
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 37 Wiki Activity

minor changes to demo script

Browse Source
This commit is contained in:
Isaac Riley 2022-02-22 16:04:11 +01:00
parent 8ff5147ee4
commit a09fe65ed2
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
vidocp/utils/deskew.py
View File

@ -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