提交 8551e07a 编写于 作者: M minqiyang

Fix flowers data read in python3

test=develop
上级 2f5a7cc4
...@@ -126,9 +126,9 @@ def reader_creator(data_file, ...@@ -126,9 +126,9 @@ def reader_creator(data_file,
batch = pickle.load(f) batch = pickle.load(f)
else: else:
batch = pickle.load(f, encoding='bytes') batch = pickle.load(f, encoding='bytes')
data = batch['data'] data = batch[six.b('data')]
labels = batch['label'] labels = batch[six.b('label')]
for sample, label in zip(data, batch['label']): for sample, label in zip(data, batch[six.b('label')]):
yield sample, int(label) - 1 yield sample, int(label) - 1
if not cycle: if not cycle:
break break
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册