diff --git a/ppcls/data/dataset/multilabel_dataset.py b/ppcls/data/dataset/multilabel_dataset.py index 2a79735086017729c99ba2ee989a75827d135fb3..45dec10e32e6446fd1e85b5c2821ec499b6cbcb3 100644 --- a/ppcls/data/dataset/multilabel_dataset.py +++ b/ppcls/data/dataset/multilabel_dataset.py @@ -59,11 +59,8 @@ class MultiLabelDataset(CommonDataset): if self._transform_ops: img = transform(img, self._transform_ops) img = img.transpose((2, 0, 1)) - label = np.array(self.labels[idx]).astype("float32") - return (img, label) - except Exception as ex: logger.error("Exception occured when parse line: {} with msg: {}". format(self.images[idx], ex))