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

refine doc-string for ImageNetDataset

Browse Source
This commit is contained in:
HydrogenSulfate 2022-08-25 14:40:19 +08:00
parent 14088d4502
commit 73e1d36697
1 changed files with 9 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
ppcls/data/dataloader/imagenet_dataset.py
View File

@ -21,21 +21,21 @@ from .common_dataset import CommonDataset
class ImageNetDataset(CommonDataset): class ImageNetDataset(CommonDataset):
"""ImageNetDataset
Args:
image_root (str): image root, path to `ILSVRC2012`
cls_label_path (str): path to annotation file `train_list.txt` or 'val_list.txt`
transform_ops (list, optional): list of transform op(s). Defaults to None.
delimiter (str, optional): delimiter. Defaults to None.
relabel (bool, optional): whether do relabel when original label do not starts from 0 or are discontinuous. Defaults to False.
"""
def __init__(self, def __init__(self,
image_root, image_root,
cls_label_path, cls_label_path,
transform_ops=None, transform_ops=None,
delimiter=None, delimiter=None,
relabel=False): relabel=False):
"""ImageNetDataset
Args:
image_root (str): _description_
cls_label_path (str): _description_
transform_ops (list, optional): list of transform op(s). Defaults to None.
delimiter (str, optional): delimiter. Defaults to None.
relabel (bool, optional): whether do relabel when original label do not starts from 0 or are discontinuous. Defaults to False.
"""
self.delimiter = delimiter if delimiter is not None else " " self.delimiter = delimiter if delimiter is not None else " "
self.relabel = relabel self.relabel = relabel
super(ImageNetDataset, self).__init__(image_root, cls_label_path, super(ImageNetDataset, self).__init__(image_root, cls_label_path,