未验证 提交 a59ac4fb 编写于 作者: B Bin Lu 提交者: GitHub

Merge pull request #823 from PaddlePaddle/Intsigstephon-patch-1

Update icartoon_dataset.py
...@@ -35,6 +35,6 @@ class ICartoonDataset(CommonDataset): ...@@ -35,6 +35,6 @@ class ICartoonDataset(CommonDataset):
np.random.shuffle(lines) np.random.shuffle(lines)
for l in lines: for l in lines:
l = l.strip().split("\t") l = l.strip().split("\t")
self.images.append(os.path.join(self._img_root, l[0][2:])) self.images.append(os.path.join(self._img_root, l[0]))
self.labels.append(int(l[1])) self.labels.append(int(l[1]))
assert os.path.exists(self.images[-1]) assert os.path.exists(self.images[-1])
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册