Skip to content

Commit 2fa4e35

Browse files
fix: count category-balanced capped images only for matching strategies
1 parent 382a128 commit 2fa4e35

1 file changed

Lines changed: 2 additions & 1 deletion

File tree

nvflow/recipes/multimodal/utils/hopchain_sample_instance_combinations.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -605,7 +605,8 @@ def main() -> None:
605605
)
606606
if was_capped:
607607
capped_images += 1
608-
category_balanced_images += 1
608+
if args.selection_strategy.startswith("balanced_by_category"):
609+
category_balanced_images += 1
609610
instances, deduplicated_count = deduplicate_instances_by_bbox_iou(
610611
instances,
611612
iou_threshold=args.iou_dedup_threshold,

0 commit comments

Comments
 (0)