未验证 提交 47d0927f 编写于 作者: D Double_V 提交者: GitHub

Merge pull request #2859 from WenmuZhou/fix_dataset

fix lable parse error when \r in the end of line
...@@ -74,7 +74,7 @@ class SimpleDataSet(Dataset): ...@@ -74,7 +74,7 @@ class SimpleDataSet(Dataset):
data_line = self.data_lines[file_idx] data_line = self.data_lines[file_idx]
try: try:
data_line = data_line.decode('utf-8') data_line = data_line.decode('utf-8')
substr = data_line.strip("\n").split(self.delimiter) substr = data_line.strip("\n").strip("\r").split(self.delimiter)
file_name = substr[0] file_name = substr[0]
label = substr[1] label = substr[1]
img_path = os.path.join(self.data_dir, file_name) img_path = os.path.join(self.data_dir, file_name)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册