From 0bf9ec3911931c6e1d337006c73c767dd91693d3 Mon Sep 17 00:00:00 2001 From: Vishal Chaurasiya Date: Wed, 4 Jun 2025 20:49:28 +0530 Subject: [PATCH 1/2] added class_dist --- supervision/dataset/utils.py | 11 ++++++++++- 1 file changed, 10 insertions(+), 1 deletion(-) diff --git a/supervision/dataset/utils.py b/supervision/dataset/utils.py index 6c30eeab0b..469c72d933 100644 --- a/supervision/dataset/utils.py +++ b/supervision/dataset/utils.py @@ -48,7 +48,16 @@ def approximate_mask_with_polygons( approximate_polygon(polygon=polygon, percentage=approximation_percentage) for polygon in polygons ] - + +def get_class_distribution(dataset: "DetectionDataset") -> Dict[str, int]: + """ + Returns a dictionary with class names as keys and sample counts as values. + """ + from collections import Counter + all_classes = [] + for det in dataset.annotations: + all_classes.extend(det['class_names']) + return dict(Counter(all_classes)) def merge_class_lists(class_lists: List[List[str]]) -> List[str]: unique_classes = set() From 6bfee366cc05ee7548affac369fc68d8cf70fdc8 Mon Sep 17 00:00:00 2001 From: "pre-commit-ci[bot]" <66853113+pre-commit-ci[bot]@users.noreply.github.com> Date: Wed, 4 Jun 2025 15:22:43 +0000 Subject: [PATCH 2/2] =?UTF-8?q?fix(pre=5Fcommit):=20=F0=9F=8E=A8=20auto=20?= =?UTF-8?q?format=20pre-commit=20hooks?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- supervision/dataset/utils.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/supervision/dataset/utils.py b/supervision/dataset/utils.py index 469c72d933..2264df3aee 100644 --- a/supervision/dataset/utils.py +++ b/supervision/dataset/utils.py @@ -48,17 +48,20 @@ def approximate_mask_with_polygons( approximate_polygon(polygon=polygon, percentage=approximation_percentage) for polygon in polygons ] - + + def get_class_distribution(dataset: "DetectionDataset") -> Dict[str, int]: """ Returns a dictionary with class names as keys and sample counts as values. """ from collections import Counter + all_classes = [] for det in dataset.annotations: - all_classes.extend(det['class_names']) + all_classes.extend(det["class_names"]) return dict(Counter(all_classes)) + def merge_class_lists(class_lists: List[List[str]]) -> List[str]: unique_classes = set()