diff --git a/ppcls/data/dataset/multilabel_dataset.py b/ppcls/data/dataset/multilabel_dataset.py index c11555003a4d51a719d8d73892bbc4e835fbc9db..2a79735086017729c99ba2ee989a75827d135fb3 100644 --- a/ppcls/data/dataset/multilabel_dataset.py +++ b/ppcls/data/dataset/multilabel_dataset.py @@ -30,9 +30,10 @@ from ppcls.data import preprocess from ppcls.data.preprocess import transform from ppcls.utils import logger +from .common_dataset import CommonDataset -class MultiLabelDataset(Dataset): +class MultiLabelDataset(CommonDataset): def _load_anno(self): assert os.path.exists(self._cls_path)