Explorar o código

fix:当h恰好是RESOLUTION_GROUP_STRIDE的倍数时,会错误地向上取整到下一个倍数。

max hai 2 semanas
pai
achega
1fd357dd97
Modificáronse 1 ficheiros con 2 adicións e 2 borrados
  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)