From 56b4d8b889da9d21083a7007f3b170252d5dd3b4 Mon Sep 17 00:00:00 2001 From: Felix <freehuang100@163.com> Date: Mon, 31 May 2021 15:49:57 +0800 Subject: [PATCH] Update multilabel_dataset.py --- ppcls/data/dataset/multilabel_dataset.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/ppcls/data/dataset/multilabel_dataset.py b/ppcls/data/dataset/multilabel_dataset.py index c1155500..2a797350 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) -- GitLab