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