提交 b541cc12 编写于 作者: 文幕地方's avatar 文幕地方

fix shape unmatch error

上级 8d841440
...@@ -58,6 +58,8 @@ class PSEPostProcess(object): ...@@ -58,6 +58,8 @@ class PSEPostProcess(object):
kernels = (pred > self.thresh).astype('float32') kernels = (pred > self.thresh).astype('float32')
text_mask = kernels[:, 0, :, :] text_mask = kernels[:, 0, :, :]
text_mask = paddle.unsqueeze(text_mask, axis=1)
kernels[:, 0:, :, :] = kernels[:, 0:, :, :] * text_mask kernels[:, 0:, :, :] = kernels[:, 0:, :, :] * text_mask
score = score.numpy() score = score.numpy()
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册