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 bug loading IN1k dataset. (#1641)

Browse Source
This commit is contained in:
Ezra-Yu 2023-06-16 15:35:27 +08:00 committed by GitHub
parent 7581b76233
commit 93e0f107c4
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
mmpretrain/datasets/imagenet.py
View File

@ -117,7 +117,7 @@ class ImageNet(CustomDataset):
if ann_file == '': if ann_file == '':
_ann_path = fileio.join_path(data_root, 'meta', f'{split}.txt') _ann_path = fileio.join_path(data_root, 'meta', f'{split}.txt')
if fileio.exists(_ann_path): if fileio.exists(_ann_path):
ann_file = _ann_path ann_file = fileio.join_path('meta', f'{split}.txt')
super().__init__( super().__init__(
data_root=data_root, data_root=data_root,
@ -210,7 +210,7 @@ class ImageNet21k(CustomDataset):
if not ann_file: if not ann_file:
_ann_path = fileio.join_path(data_root, 'meta', f'{split}.txt') _ann_path = fileio.join_path(data_root, 'meta', f'{split}.txt')
if fileio.exists(_ann_path): if fileio.exists(_ann_path):
ann_file = _ann_path ann_file = fileio.join_path('meta', f'{split}.txt')
logger = MMLogger.get_current_instance() logger = MMLogger.get_current_instance()