Merge pull request #371 from sbrunner/renovate/scikit-image

Fix for already gray image
sbrunner · Mar 8, 2023 · a5136d5 · a5136d5
2 parents 044e2ab + 5974ad0
commit a5136d5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/deskew/cli.py b/deskew/cli.py
@@ -24,7 +24,7 @@ def main() -> None:
     options = parser.parse_args()
 
     image = io.imread(options.input)
-    grayscale = image if len(image) == 2 else rgb2gray(image)
+    grayscale = image if len(image.shape) == 2 else rgb2gray(image)
     angle = determine_skew(grayscale, sigma=options.sigma, num_peaks=options.num_peaks)
     if options.output is None:
         print(f"Estimated angle: {angle}")