mirrors/PaddleOCR
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/PaddleOCR.git synced 2025-06-03 21:53:39 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix kie mode will crash when it has no kie_cls in label info

Browse Source
This commit is contained in:
HinGwenWoong 2022-05-05 10:11:21 +08:00
parent cb9a8c4b96
commit 9e04c09f62
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
PPOCRLabel/PPOCRLabel.py
View File

@ -1439,7 +1439,7 @@ class MainWindow(QMainWindow):
DEFAULT_LOCK_COLOR, key_cls, box['difficult'])) DEFAULT_LOCK_COLOR, key_cls, box['difficult']))
if imgidx in self.PPlabel.keys(): if imgidx in self.PPlabel.keys():
for box in self.PPlabel[imgidx]: for box in self.PPlabel[imgidx]:
key_cls = None if not self.kie_mode else box['key_cls'] key_cls = None if not self.kie_mode else box.get('key_cls', 'None')
shapes.append((box['transcription'], box['points'], None, key_cls, box.get('difficult', False))) shapes.append((box['transcription'], box['points'], None, key_cls, box.get('difficult', False)))
self.loadLabels(shapes) self.loadLabels(shapes)
@ -1584,7 +1584,7 @@ class MainWindow(QMainWindow):
for image, info in label_dict.items(): for image, info in label_dict.items():
for box in info: for box in info:
if "key_cls" not in box: if "key_cls" not in box:
continue box.update({"key_cls": "None"})
self.existed_key_cls_set.add(box["key_cls"]) self.existed_key_cls_set.add(box["key_cls"])
if len(self.existed_key_cls_set) > 0: if len(self.existed_key_cls_set) > 0:
for key_text in self.existed_key_cls_set: for key_text in self.existed_key_cls_set: