Skip to content

GitLab

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

fix error (#1177)

parent bc563c64
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
ppocr/data/simple_dataset.py
View file @ 169b629b
...@@ -104,8 +104,8 @@ class SimpleDataSet(Dataset): ...@@ -104,8 +104,8 @@ class SimpleDataSet(Dataset):
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").split(self.delimiter)
file_name = substr[1] file_name = substr[0]
label = substr[0] label = substr[1]
img_path = os.path.join(self.data_dir, file_name) img_path = os.path.join(self.data_dir, file_name)
data = {'img_path': img_path, 'label': label} data = {'img_path': img_path, 'label': label}
with open(data['img_path'], 'rb') as f: 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