From 6dc3f7bd7f5395ebbf7ec6bc2bf76c3b47862937 Mon Sep 17 00:00:00 2001
From: uni19 <csljxing@gmail.com>
Date: Fri, 20 Aug 2021 02:42:21 +0800
Subject: [PATCH] ensure items in dataset have the same order across multi
machine (#780)
---
mmseg/datasets/custom.py | 1 +
1 file changed, 1 insertion(+)
diff --git a/mmseg/datasets/custom.py b/mmseg/datasets/custom.py
index 719ca013..a86fabb9 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
--
GitLab