fix lable parse error when \r in the end of line

2026-09-21 05:12:13 +08:00 · 2021-05-21 12:02:03 +08:00
parent 96fd27c430
commit 9497580dce
1 changed files with 1 additions and 1 deletions
@@ -74,7 +74,7 @@ class SimpleDataSet(Dataset):
        data_line = self.data_lines[file_idx]
        try:
            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]
            label = substr[1]
            img_path = os.path.join(self.data_dir, file_name)