首页 PaddleOCR 帖子详情
调用OCR包推理图片的错误
收藏
快速回复
PaddleOCR 问答OCR 946 3
调用OCR包推理图片的错误
收藏
快速回复
PaddleOCR 问答OCR 946 3

代码:

ocr = PaddleOCR(use_angle_cls=True, use_gpu=False)

img_path = '/home/jinrm/OCR_Model_7/PaddleOCR-release-2.3 2/img_path/Image_00002.jpg' # 这个是自己的图片,自行放置在代码目录下修改名称

result = ocr.ocr(img_path, cls=True)
for line in result:
print(line)

error log:

7 result = ocr.ocr(img_path, cls=True)
8 for line in result:
9 print(line)

/home/jinrm/OCR_Model_7/PaddleOCR-release-2.3 2/paddleocr.py in ocr(self, img, det, rec, cls)
341 img = cv2.cvtColor(img, cv2.COLOR_GRAY2BGR)
342 if det and rec:
--> 343 dt_boxes, rec_res = self.__call__(img, cls)
344 return [[box.tolist(), res] for box, res in zip(dt_boxes, rec_res)]
345 elif det and not rec:

/home/jinrm/OCR_Model_7/PaddleOCR-release-2.3 2/tools/infer/predict_system.py in __call__(self, img, cls)
79 len(img_crop_list), elapse))
80
---> 81 rec_res, elapse = self.text_recognizer(img_crop_list)
82 logger.debug("rec_res num : {}, elapse : {}".format(
83 len(rec_res), elapse))

/home/jinrm/OCR_Model_7/PaddleOCR-release-2.3 2/tools/infer/predict_rec.py in __call__(self, img_list)
261 self.autolog.times.stamp()
262 preds = outputs[0]
--> 263 rec_result = self.postprocess_op(preds)
264 for rno in range(len(rec_result)):
265 rec_res[indices[beg_img_no + rno]] = rec_result[rno]

/home/jinrm/OCR_Model_7/PaddleOCR-release-2.3 2/ppocr/postprocess/rec_postprocess.py in __call__(self, preds, label, *args, **kwargs)
113 if isinstance(preds, paddle.Tensor):
114 preds = preds.numpy()
--> 115 preds_idx = preds.argmax(axis=2)
116 preds_prob = preds.max(axis=2)
117 text = self.decode(preds_idx, preds_prob, is_remove_duplicate=True)

AxisError: axis 2 is out of bounds for array of dimension 2

0
收藏
回复
全部评论(3)
时间顺序
FutureSI
#2 回复于2021-09

preds张量的形状错误,打印下preds的shape找找原因

0
回复
FutureSI
#3 回复于2021-09

或者是不是preds.argmax(axis=2)的axis参数错了,维度是从0开始算的。你的preds张量时2维的。

0
回复
划飞桨的Daniel
#4 回复于2021-09

建议还是先跑跑官方提供的例子,图片应该也有个压缩包可以下载,先跑通官方的例子,再查问题就好查了

 

0
回复
需求/bug反馈?一键提issue告诉我们
发现bug?如果您知道修复办法,欢迎提pr直接参与建设飞桨~
在@后输入用户全名并按空格结束,可艾特全站任一用户