未验证 提交 bb1e4b9e 编写于 作者: W whs 提交者: GitHub

Remove label from infer reader. (#3882)

上级 7b0e8b41
......@@ -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.
先完成此消息的编辑!
想要评论请 注册