|
|
@@ -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)
|
|
|
|