[PaddlePaddle/PaddleOCR]paddle怎么训练docunet?

2024-05-13 6 views
5

请提供下述完整信息以便快速定位问题/Please provide the following information to quickly locate the problem

  • 系统环境/System Environment:
  • 版本号/Version:Paddle: PaddleOCR:2.7.1 问题相关组件/Related components:
  • 运行指令/Command Code:
  • 完整报错/Complete Error Message:

请尽量不要包含图片在问题中/Please try to not include the image in the issue.

https://github.com/teresasun/docUnet.pytorch

回答

3

谢谢。 有训练好的模型吗?怎么使用doctr++

7

暂时没有。使用方法可以参考该PR。由于暂未合入,开发者提交的doctr++代码是否完备以及是否完全能复现,暂时还未知。

8

是python实现的,与操作系统无关。

7

报这个错误,怎么解决?

python predict.py -i "F:\0Table03\3.jpg" -m L:\img2table\DocTrPP-main\best.ckpt -o F:\3result.jpg Namespace(image='F:\0Table03\3.jpg', model='L:\img2table\DocTrPP-main\best.ckpt', output='F:\3result.jpg') W0427 08:19:53.655071 50756 gpu_resources.cc:61] Please NOTE: device: 0, GPU Compute Capability: 6.1, Driver API Version: 12.0, Runtime API Version: 10.2 W0427 08:19:53.673440 50756 gpu_resources.cc:91] device: 0, cuDNN Version: 7.6. Traceback (most recent call last): File "predict.py", line 69, in run(args) File "predict.py", line 25, in run bm = model(x) File "D:\Python3.8.8\lib\site-packages\paddle\fluid\dygraph\layers.py", line 930, in call return self._dygraph_call_func(*inputs, kwargs) File "D:\Python3.8.8\lib\site-packages\paddle\fluid\dygraph\layers.py", line 915, in _dygraph_call_func outputs = self.forward(*inputs, *kwargs) File "L:\img2table\DocTrPP-main\GeoTr.py", line 370, in forward fmap1 = self.getattr(self.encoder_block[0])(fmap) File "D:\Python3.8.8\lib\site-packages\paddle\fluid\dygraph\layers.py", line 930, in call return self._dygraph_call_func(inputs, kwargs) File "D:\Python3.8.8\lib\site-packages\paddle\fluid\dygraph\layers.py", line 915, in _dygraph_call_func outputs = self.forward(*inputs, kwargs) File "L:\img2table\DocTrPP-main\GeoTr.py", line 206, in forward pos = self.position_embedding( File "D:\Python3.8.8\lib\site-packages\paddle\fluid\dygraph\layers.py", line 930, in call return self._dygraph_call_func(*inputs, *kwargs) File "D:\Python3.8.8\lib\site-packages\paddle\fluid\dygraph\layers.py", line 915, in _dygraph_call_func outputs = self.forward(inputs, kwargs) File "L:\img2table\DocTrPP-main\position_encoding.py", line 55, in forward dim_t = self.temperature * (2 (dim_t // 2) / self.num_pos_feats) File "D:\Python3.8.8\lib\site-packages\paddle\fluid\dygraph\math_op_patch.py", line 299, in impl return math_op(self, other_var, 'axis', axis) RuntimeError: (NotFound) There are no kernels which are registered in the elementwise_floordiv operator. [Hint: Expected kernels_iter != all_op_kernels.end(), but received kernels_iter == all_op_kernels.end().] (at ..\paddle\fluid\imperative\prepared_operator.cc:341) [operator < elementwise_floordiv > error]

0

要哪个版本?requirements里没看到要paddlepaddle 版本 hdf5storage loguru numpy scipy opencv-python matplotlib albumentations

1

最新版paddle就好了

7

哦。我装的是paddlepaddle-gpu 2.3.2 cuda10.2

0

有没有办法转换成onnx格式使用?怎么用?

6

问题非paddleocr问题,此处关闭该issue,可在具体仓库中重开issue提问