mirrors
/
mmpretrain
mirror of https://github.com/open-mmlab/mmpretrain.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

[Fix] Fix bug loading IN1k dataset. (#1641)

pull/1651/head
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 == '':
_ann_path = fileio.join_path(data_root, 'meta', f'{split}.txt')
if fileio.exists(_ann_path):
ann_file = _ann_path
ann_file = fileio.join_path('meta', f'{split}.txt')
super().__init__(
data_root=data_root,
@ -210,7 +210,7 @@ class ImageNet21k(CustomDataset):
if not ann_file:
_ann_path = fileio.join_path(data_root, 'meta', f'{split}.txt')
if fileio.exists(_ann_path):
ann_file = _ann_path
ann_file = fileio.join_path('meta', f'{split}.txt')
logger = MMLogger.get_current_instance()