change label filename

Evezerest · May 6, 2022 · b657a3a · b657a3a
1 parent f1d0c8a
commit b657a3a
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/PPOCRLabel/PPOCRLabel.py b/PPOCRLabel/PPOCRLabel.py
@@ -2382,7 +2382,7 @@ def exportJSON(self):
 
  # save dict
  html = {'structure': {'tokens': token_list}, 'cell': cells}
- json_results.append({'filename': filename, 'split': split, 'imgid': imgid, 'html': html})
+ json_results.append({'filename': os.path.basename(image_path), 'split': split, 'imgid': imgid, 'html': html})
  imgid += 1
 
  # save json

diff --git a/PPOCRLabel/libs/utils.py b/PPOCRLabel/libs/utils.py
@@ -225,7 +225,7 @@ def convert_token(html_list):
  if 'rowspan' in col:
  _, n = col.split('rowspan=')
  token_list.append(" rowspan=\"{}\"".format(n))
- token_list.append(">")
+ token_list.extend([">", "</td>"])
  token_list.append("</tr>")
  token_list.append("</tbody>")