提交 02621d3f 编写于 作者: qq_25193841's avatar qq_25193841

Fix a bug of kie as null

Fix a bug of kie as null
上级 2954d396
...@@ -1471,7 +1471,7 @@ class MainWindow(QMainWindow): ...@@ -1471,7 +1471,7 @@ class MainWindow(QMainWindow):
# box['ratio'] of the shapes saved in lockedShapes contains the ratio of the # box['ratio'] of the shapes saved in lockedShapes contains the ratio of the
# four corner coordinates of the shapes to the height and width of the image # four corner coordinates of the shapes to the height and width of the image
for box in self.canvas.lockedShapes: for box in self.canvas.lockedShapes:
key_cls = None if not self.kie_mode else box['key_cls'] key_cls = 'None' if not self.kie_mode else box['key_cls']
if self.canvas.isInTheSameImage: if self.canvas.isInTheSameImage:
shapes.append((box['transcription'], [[s[0] * width, s[1] * height] for s in box['ratio']], shapes.append((box['transcription'], [[s[0] * width, s[1] * height] for s in box['ratio']],
DEFAULT_LOCK_COLOR, key_cls, box['difficult'])) DEFAULT_LOCK_COLOR, key_cls, box['difficult']))
...@@ -1480,7 +1480,7 @@ class MainWindow(QMainWindow): ...@@ -1480,7 +1480,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.get('key_cls', 'None') 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)
...@@ -2266,7 +2266,7 @@ class MainWindow(QMainWindow): ...@@ -2266,7 +2266,7 @@ class MainWindow(QMainWindow):
rec_res = self.ocr.ocr(patch, det=False, rec=True, cls=False) rec_res = self.ocr.ocr(patch, det=False, rec=True, cls=False)
text = rec_res[0][0] text = rec_res[0][0]
if text != '': if text != '':
texts += text + (' ' if text[0].isalpha() else '') # add space between english word texts += text + ('' if text[0].isalpha() else ' ') # add space between english word
probs += rec_res[0][1] probs += rec_res[0][1]
probs = probs / len(bboxes) probs = probs / len(bboxes)
result = [(texts.strip(), probs)] result = [(texts.strip(), probs)]
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册