未验证 提交 d2630c3a 编写于 作者: F Felix 提交者: GitHub

Update multilabel_dataset.py

上级 56b4d8b8
......@@ -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))
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册