mirrors/mmengine
1
0
Fork 0
mirror of https://github.com/open-mmlab/mmengine.git synced 2025-06-03 21:54:44 +08:00
Code Issues Projects Releases Wiki Activity

remove ann_file arg in load_ann_list (#211)

Browse Source
This commit is contained in:
Tao Gong 2022-04-27 19:47:49 +08:00 committed by GitHub
parent 17dbac1812
commit 8668c14d25
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 7 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
mmengine/dataset/base_dataset.py
View File

@ -293,7 +293,7 @@ class BaseDataset(Dataset):
if self._fully_initialized:
return
# load data information
self.data_list = self.load_data_list(self.ann_file)
self.data_list = self.load_data_list()
# filter illegal data, such as data that has no annotations.
self.data_list = self.filter_data()
# Get subset data according to indices.
@ -408,8 +408,8 @@ class BaseDataset(Dataset):
raise Exception(f'Cannot find valid image after {self.max_refetch}! '
'Please check your image path and pipeline')
def load_data_list(self, ann_file: str) -> List[dict]:
"""Load annotations from an annotation file.
def load_data_list(self) -> List[dict]:
"""Load annotations from an annotation file named as ``self.ann_file``
If the annotation file does not follow `OpenMMLab 2.0 format dataset
<https://github.com/open-mmlab/mmengine/blob/main/docs/zh_cn/tutorials/basedataset.md>`_ .
@ -417,15 +417,13 @@ class BaseDataset(Dataset):
information of annotation file will be overwritten :attr:`METAINFO`
and ``metainfo`` argument of constructor.
Args:
ann_file (str): Absolute annotation file path if ``self.root=None``
or relative path if ``self.root=/path/to/data/``.
Returns:
list[dict]: A list of annotation.
""" # noqa: E501
check_file_exist(ann_file)
annotations = load(ann_file)
# `self.ann_file` denotes the absolute annotation file path if
# `self.root=None` or relative path if `self.root=/path/to/data/`.
check_file_exist(self.ann_file)
annotations = load(self.ann_file)
if not isinstance(annotations, dict):
raise TypeError(f'The annotations loaded from annotation file '
f'should be a dict, but got {type(annotations)}!')