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

Merge branch 'xiexinchen/fix_dataset' into 'refactor_dev'

Browse Source 
[Refactor] Fix load annotations by file

See merge request openmmlab-enterprise/openmmlab-ce/mmsegmentation!62
This commit is contained in:
zhengmiao 2022-07-11 11:31:38 +00:00
commit b08e0bcc44
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
mmseg/datasets/custom.py
View File

@ -233,13 +233,15 @@ class CustomDataset(BaseDataset):
list[dict]: All data info of dataset.
"""
data_list = []
img_dir = self.data_prefix.get('img_path', None)
ann_dir = self.data_prefix.get('seg_map_path', None)
if osp.isfile(self.ann_file):
lines = mmcv.list_from_file(
self.ann_file, file_client_args=self.file_client_args)
for line in lines:
img_name = line.strip()
data_info = dict(img_path=img_name + self.img_suffix)
data_info = dict(
img_path=osp.join(img_dir, img_name + self.img_suffix))
if ann_dir is not None:
seg_map = img_name + self.seg_map_suffix
data_info['seg_map_path'] = osp.join(ann_dir, seg_map)
@ -248,7 +250,6 @@ class CustomDataset(BaseDataset):
data_info['seg_fields'] = []
data_list.append(data_info)
else:
img_dir = self.data_prefix['img_path']
for img in self.file_client.list_dir_or_file(
dir_path=img_dir,
list_dir=False,