From aace571368787cf739c51af72dc634a4b23f3fb1 Mon Sep 17 00:00:00 2001 From: Robert Sachunsky Date: Fri, 17 Jul 2026 19:15:54 +0200 Subject: [PATCH] cnn-rnn-ocr: fix --- src/eynollah/utils/utils_ocr.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/src/eynollah/utils/utils_ocr.py b/src/eynollah/utils/utils_ocr.py index d8b87b4..14cbb79 100644 --- a/src/eynollah/utils/utils_ocr.py +++ b/src/eynollah/utils/utils_ocr.py @@ -182,10 +182,11 @@ def get_orientation_moments(contour): def get_orientation_moments_of_mask(mask): mask=mask.astype('uint8') - contours, _ = cv2.findContours(mask[:,:,0], cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE) - - largest_contour = max(contours, key=cv2.contourArea) if contours else None - + contours, _ = cv2.findContours(mask, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE) + + if not len(contours): + return 0 + largest_contour = max(contours, key=cv2.contourArea) moments = cv2.moments(largest_contour) if moments["mu20"] - moments["mu02"] == 0: # Avoid division by zero return 90 if moments["mu11"] > 0 else -90