Solene Tarride
--- a/dan/ocr/predict/attention.py

+ 1

− 3
+++ b/dan/ocr/predict/attention.py

+ 1

− 3
 @@ -342,11 +342,9 @@ def get_grid_search_contour(coverage_vector, bin_mask, height=50):

    # Limit bounding box shape
    width = bin_mask.shape[1]
-    y_range = range(y_min, max(y_max - height, y_min + 1))

    best_sum_att = 0
-
-    for y in y_range:
+    for y in range(y_min, max(y_max - height, y_min + 1)):
        sum_att = coverage_vector[y : y + height, 0:width].sum()
        if sum_att > best_sum_att:
            best_sum_att = sum_att