fix coco pbars

openvinotoolkit · Jul 17, 2023 · 11ce4c5 · 11ce4c5
1 parent 5f98882
commit 11ce4c5
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/datumaro/plugins/data_formats/coco/base.py b/src/datumaro/plugins/data_formats/coco/base.py
@@ -266,7 +266,7 @@ def _stream_items(self) -> Iterator[DatasetItem]:
 
         for img_info, ann_infos in pbars.iter(
             self._page_mapper,
-            desc=f"Parsing image info in '{osp.basename(self._path)}'",
+            desc=f"Importing '{self._subset}'",
         ):
             parsed = self._parse_item(img_info)
             if parsed is None:
@@ -314,7 +314,7 @@ def _gen_ann(info_lists):
         for ann_info in pbar.iter(
             _gen_ann(ann_lists),
             desc=f"Importing '{self._subset}'",
-            total=len(img_lists),
+            total=len(ann_lists),
         ):
             try:
                 img_id = self._parse_field(ann_info, "image_id", int)