Ver Fonte

Merge pull request #3880 from baymax2099/2.6.2fix

Fix rounding error for height and width normalization
Xiaomeng Zhao há 2 semanas atrás
pai
commit
324dd75a52
1 ficheiros alterados com 2 adições e 2 exclusões
  1. 2 2
      mineru/backend/pipeline/batch_analyze.py

+ 2 - 2
mineru/backend/pipeline/batch_analyze.py

@@ -289,8 +289,8 @@ class BatchAnalyze:
                     h, w = cropped_img.shape[:2]
                     # 使用更大的分组容差,减少分组数量
                     # 将尺寸标准化到32的倍数
-                    normalized_h = ((h + RESOLUTION_GROUP_STRIDE) // RESOLUTION_GROUP_STRIDE) * RESOLUTION_GROUP_STRIDE  # 向上取整到32的倍数
-                    normalized_w = ((w + RESOLUTION_GROUP_STRIDE) // RESOLUTION_GROUP_STRIDE) * RESOLUTION_GROUP_STRIDE
+                    normalized_h = ((h + RESOLUTION_GROUP_STRIDE - 1) // RESOLUTION_GROUP_STRIDE) * RESOLUTION_GROUP_STRIDE  # 向上取整到32的倍数
+                    normalized_w = ((w + RESOLUTION_GROUP_STRIDE - 1) // RESOLUTION_GROUP_STRIDE) * RESOLUTION_GROUP_STRIDE
                     group_key = (normalized_h, normalized_w)
                     resolution_groups[group_key].append(crop_info)