ensure items in dataset have the same order across multi machine

open-mmlab · Aug 12, 2021 · f484482 · f484482
1 parent 58f5dbc
commit f484482
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/mmseg/datasets/custom.py b/mmseg/datasets/custom.py
@@ -151,6 +151,7 @@ def load_annotations(self, img_dir, img_suffix, ann_dir, seg_map_suffix,
                     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