From db856df43eb0c5552c60f11ed2b3eadece4b9ba5 Mon Sep 17 00:00:00 2001 From: pvys <55968684+pvys@users.noreply.github.com> Date: Thu, 29 Jul 2021 11:21:22 +0900 Subject: [PATCH] [Fix] Fix ImageNet dataset annotation file parse bug. (#370) Co-authored-by: mzr1996 --- mmcls/datasets/imagenet.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mmcls/datasets/imagenet.py b/mmcls/datasets/imagenet.py index 99fd1b998..64d58ed5e 100644 --- a/mmcls/datasets/imagenet.py +++ b/mmcls/datasets/imagenet.py @@ -1091,7 +1091,7 @@ class ImageNet(BaseDataset): self.folder_to_idx = folder_to_idx elif isinstance(self.ann_file, str): with open(self.ann_file) as f: - samples = [x.strip().split(' ') for x in f.readlines()] + samples = [x.strip().rsplit(' ', 1) for x in f.readlines()] else: raise TypeError('ann_file must be a str or None') self.samples = samples