未验证 提交 169b629b 编写于 作者: L littletomatodonkey 提交者: GitHub

fix error (#1177)

上级 bc563c64
......@@ -104,8 +104,8 @@ class SimpleDataSet(Dataset):
try:
data_line = data_line.decode('utf-8')
substr = data_line.strip("\n").split(self.delimiter)
file_name = substr[1]
label = substr[0]
file_name = substr[0]
label = substr[1]
img_path = os.path.join(self.data_dir, file_name)
data = {'img_path': img_path, 'label': label}
with open(data['img_path'], 'rb') as f:
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册