supervision
supervision copied to clipboard
added class_dist
Added a functionality to get the class distribution -
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))