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

Update icartoon_dataset.py

上级 15acd6a5
......@@ -35,6 +35,6 @@ class ICartoonDataset(CommonDataset):
np.random.shuffle(lines)
for l in lines:
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]))
assert os.path.exists(self.images[-1])
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册