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