Update icartoon_dataset.py
parent
c0af015025
commit
3a2f97a911
|
@ -29,10 +29,6 @@ class ICartoonDataset(CommonDataset):
|
||||||
|
|
||||||
with open(self._cls_path) as fd:
|
with open(self._cls_path) as fd:
|
||||||
lines = fd.readlines()
|
lines = fd.readlines()
|
||||||
if seed is not None:
|
|
||||||
np.random.RandomState(seed).shuffle(lines)
|
|
||||||
else:
|
|
||||||
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]))
|
self.images.append(os.path.join(self._img_root, l[0]))
|
||||||
|
|
Loading…
Reference in New Issue