ensure items in dataset have the same order across multi machine (#780)

2021-08-20 02:42:21 +08:00 · 2021-08-20 02:42:21 +08:00 · 6dc3f7bd7f
parent ebf3c084ac
commit 6dc3f7bd7f
1 changed files with 1 additions and 0 deletions
--- a/mmseg/datasets/custom.py
+++ b/mmseg/datasets/custom.py
@ -152,6 +152,7 @@ class CustomDataset(Dataset):
                    seg_map = img.replace(img_suffix, seg_map_suffix)
                    img_info['ann'] = dict(seg_map=seg_map)
                img_infos.append(img_info)
+            img_infos = sorted(img_infos, key=lambda x: x['filename'])

        print_log(f'Loaded {len(img_infos)} images', logger=get_root_logger())
        return img_infos