refactor imagenet dataset

open-mmlab · Apr 28, 2023 · b770e09 · b770e09
1 parent f80293d
commit b770e09
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/mmpretrain/datasets/imagenet.py b/mmpretrain/datasets/imagenet.py
@@ -56,6 +56,8 @@ class ImageNet(CustomDataset):
             ``ann_file``. Defaults to ''.
         split (str): The dataset split, supports "train" and "val".
             Default to ''.
+        data_prefix (str | dict): Prefix for training data. Defaults to ''.
+        ann_file (str): Annotation file path. Defaults to ''.
         metainfo (dict, optional): Meta information for dataset, such as class
             information. Defaults to None.
         **kwargs: Other keyword arguments in :class:`CustomDataset` and