Unverified Commit 169b629b authored by littletomatodonkey's avatar littletomatodonkey Committed by GitHub
Browse files

fix error (#1177)

parent 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% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment