Merge pull request #22 from tr1v1al/master

Fixing quantize to include xmax-xmin=9 case
georg-wolflein · Jan 11, 2025 · c9b11d6 · c9b11d6
2 parents cb9a42d + 5c21bf5
commit c9b11d6
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/chesscog/corner_detection/detect_corners.py b/chesscog/corner_detection/detect_corners.py
@@ -368,10 +368,10 @@ def _quantize_points(cfg: CN, warped_scaled_points: np.ndarray, intersection_poi
     ymax = row_ys.max()
 
     # Ensure we a have a maximum of 9 rows/cols
-    while xmax - xmin > 9:
+    while xmax - xmin > 8:
         xmax -= 1
         xmin += 1
-    while ymax - ymin > 9:
+    while ymax - ymin > 8:
         ymax -= 1
         ymin += 1
     col_mask = (col_xs >= xmin) & (col_xs <= xmax)