diff --git a/src/datumaro/plugins/data_formats/coco/base.py b/src/datumaro/plugins/data_formats/coco/base.py index 95ca81148c..9fece26569 100644 --- 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)