提交 881547ef 编写于 作者: W wanghaoshuang

Merge branch 'develop' of https://github.com/paddlepaddle/models into fix_slim_demo

......@@ -212,8 +212,7 @@ class DataGenerator(object):
img = img.resize((img.size[0], DATA_SHAPE[1])) # resize height
img = np.array(img) - 127.5
img = img[np.newaxis, ...]
label = [int(c) for c in line.split(' ')[3].split(',')]
yield img, label
yield img, [[0]]
if img_label_list is not None:
lines = []
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册