move fixing of class_ids to load_pascal_voc_annotations

supervision/dataset/core.py

@@ -241,10 +241,6 @@ class DetectionDataset(BaseDataset):
             force_masks=force_masks,
         )
 
-        for annotation in annotations.values():
-            class_id = [classes.index(class_name) for class_name in annotation.class_id]
-            annotation.class_id = np.array(class_id)
-
         return DetectionDataset(classes=classes, images=images, annotations=annotations)
 
     @classmethod

supervision/dataset/formats/pascal_voc.py

@@ -203,6 +203,10 @@ def load_pascal_voc_annotations(
 
     classes = list(set(classes))
 
+    for annotation in annotations.values():
+        class_id = [classes.index(class_name) for class_name in annotation.class_id]
+        annotation.class_id = np.array(class_id)
+
     return classes, images, annotations