mirrors/mmclassification
1
0
Fork 0
mirror of https://github.com/open-mmlab/mmclassification.git synced 2025-06-03 21:53:55 +08:00
Code Issues Projects Releases Wiki Activity

[Fix] Fix ImageNet dataset annotation file parse bug. (#370)

Browse Source 
Co-authored-by: mzr1996 <mzr1996@163.com>
This commit is contained in:
pvys 2021-07-29 11:21:22 +09:00 committed by GitHub
parent c92eefbb4a
commit db856df43e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mmcls/datasets/imagenet.py
View File

@ -1091,7 +1091,7 @@ class ImageNet(BaseDataset):
self.folder_to_idx = folder_to_idx self.folder_to_idx = folder_to_idx
elif isinstance(self.ann_file, str): elif isinstance(self.ann_file, str):
with open(self.ann_file) as f: 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: else:
raise TypeError('ann_file must be a str or None') raise TypeError('ann_file must be a str or None')
self.samples = samples self.samples = samples