fix error (#1177)

2026-09-21 13:23:18 +08:00 · 2020-11-21 23:02:18 +08:00
parent bc563c642c
commit 169b629b0f
1 changed files with 2 additions and 2 deletions
@@ -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: