diff --git a/nvflow/recipes/multimodal/utils/hopchain_sample_instance_combinations.py b/nvflow/recipes/multimodal/utils/hopchain_sample_instance_combinations.py index 3517e0c..e7c26fe 100644 --- a/nvflow/recipes/multimodal/utils/hopchain_sample_instance_combinations.py +++ b/nvflow/recipes/multimodal/utils/hopchain_sample_instance_combinations.py @@ -605,7 +605,8 @@ def main() -> None: ) if was_capped: capped_images += 1 - category_balanced_images += 1 + if args.selection_strategy.startswith("balanced_by_category"): + category_balanced_images += 1 instances, deduplicated_count = deduplicate_instances_by_bbox_iou( instances, iou_threshold=args.iou_dedup_threshold,