Update supervision/dataset/formats/coco.py
Co-authored-by: Piotr Skalski <piotr.skalski92@gmail.com>
This commit is contained in:
1 parent
4414e15a00
commit
d543be924f
1 file changed
+1
-2
@@ -179,8 +179,7 @@ def get_coco_class_index_mapping(annotations_path: str) -> Dict[int, int]:
|
||||
class_mapping = build_coco_class_index_mapping(
|
||||
coco_categories=coco_data["categories"], target_classes=classes
|
||||
)
|
||||
inv_class_mapping = {v: k for k, v in class_mapping.items()}
|
||||
return inv_class_mapping
|
||||
return {v: k for k, v in class_mapping.items()}
|
||||
|
||||
|
||||
def load_coco_annotations(
|
||||
|
||||
Reference in new issue
Block a user