paddle模型加载报错

[Sevenal]

代码示例

 def table_detect(self, img_path, **kwargs):
        """表格检测，检测图片上的表格"""
        
        table_det = TableDetector(
            obj_model_path = "./TableParse/table_det_models/obj_det_paddle",
            edge_model_path = "./TableParse/table_det_models/edge_det_paddle",
            cls_model_path = "./TableParse/table_det_models/cls_det_paddle",
            use_obj_det=True,
            use_edge_det=True,
            use_cls_det=True
        )
        
        # elapse 为检测边界信息
        result, elapse = table_det(img_path)
        obj_det_elapse, edge_elapse, rotate_det_elapse = elapse
        
        img = img_loader(img_path)
        img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
        file_name_with_ext = os.path.basename(img_path)
        file_name, file_ext = os.path.splitext(file_name_with_ext)
        
        out_dir = kwargs["out_dir"] if "out_dir" in kwargs else "outputs"
        if not os.path.exists(out_dir):
            os.makedirs(out_dir)
        
        extract_img = img.copy()
        
        for i, res in enumerate(result):
            box = res["box"]
            lt, rt, rb, lb = res["lt"], res["rt"], res["rb"], res["lb"]
            # 带识别框和左上角方向位置
            img = visuallize(img, box, lt, rt, rb, lb)
            # 透视变换提取表格图片
            wrapped_img = extract_table_img(extract_img.copy(), lt, rt, rb, lb)
            cv2.imwrite(f"{out_dir}/{file_name}_table_{i}.jpg", wrapped_img)
        
        visual = kwargs["visual"] if "visual" in kwargs else False
        if visual:
            cv2.imwrite(f"{out_dir}/{file_name}_visualize.jpg", img)

## 目录结构
![image](https://github.com/user-attachments/assets/105240ea-b37f-42d7-b435-63a02d21a9df)

## 报错信息
Traceback (most recent call last):
  File "d:\Work\主动元数据-解析\bin\TableParse\table_parse.py", line 62, in <module>
    table_parse.table_detect(r"assets\image\Snipaste_2025-01-08_10-19-08.png", visual=True)
  File "d:\Work\主动元数据-解析\bin\TableParse\table_parse.py", line 22, in table_detect
    table_det = TableDetector(
                ^^^^^^^^^^^^^^
  File "D:\Anaconda\envs\NLP\Lib\site-packages\rapid_table_det_paddle\inference.py", line 23, in __init__
    self.obj_detector = ObjectDetector(obj_model_path)
                        ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "D:\Anaconda\envs\NLP\Lib\site-packages\rapid_table_det_paddle\predictor.py", line 14, in __init__
    self.model = paddle.jit.load(model_path)
                 ^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "D:\Anaconda\envs\NLP\Lib\site-packages\decorator.py", line 232, in fun
    return caller(func, *(extras + args), **kw)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "D:\Anaconda\envs\NLP\Lib\site-packages\paddle\base\wrapped_decorator.py", line 26, in __impl__
    return wrapped_func(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "D:\Anaconda\envs\NLP\Lib\site-packages\paddle\base\framework.py", line 593, in __impl__
    return func(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^
  File "D:\Anaconda\envs\NLP\Lib\site-packages\paddle\jit\api.py", line 1584, in load
    return TranslatedLayer._construct(model_path, config)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "D:\Anaconda\envs\NLP\Lib\site-packages\decorator.py", line 232, in fun
    return caller(func, *(extras + args), **kw)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "D:\Anaconda\envs\NLP\Lib\site-packages\paddle\base\wrapped_decorator.py", line 26, in __impl__
    return wrapped_func(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "D:\Anaconda\envs\NLP\Lib\site-packages\paddle\base\framework.py", line 593, in __impl__
    return func(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^^^
  File "D:\Anaconda\envs\NLP\Lib\site-packages\paddle\jit\translated_layer.py", line 1444, in _construct
    persistable_vars = _construct_params_and_buffers(
                       ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "D:\Anaconda\envs\NLP\Lib\site-packages\paddle\jit\translated_layer.py", line 875, in _construct_params_and_buffers
    model_path, programs['forward'], params_filename
                ~~~~~~~~^^^^^^^^^^^
KeyError: 'forward'

[Joker1212]

能给一下paddlepaddle的版本信息吗？看起来好像是它版本的问题

[Sevenal]

这是我安装的版本
paddleocr 2.7.3
paddlepaddle-gpu 2.6.1

[Joker1212]

我试了paddle的cpu2.6.2和gpu的2.6.2版本，能正常运行，另外，模型是需要下载到本地指定目录，且命名对齐,如果是环境影响可以conda新建一个环境试试呢？