diff --git a/mmseg/datasets/custom.py b/mmseg/datasets/custom.py index 719ca013f40caf06f89e196fdf8d2e4fe2ecc9c7..a86fabb9722eab2546a54421948797cec1a94f8d 100644 --- 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