Skip to content

自训练 PP-OCRv2 识别模型转 Lite .nb 替换后 Android 报 9088!==128,与文档“可替换模型”不符,请求提供与官方 .nb 同图结构的 pdmodel 或导出脚本 #10707

@yongfu1

Description

@yongfu1

环境:

  • PaddleOCR:main / release/2.6 均试过
  • Paddle-Lite:2.10(与 Android 端一致)
  • 操作:自训练 ch_PP-OCRv2 rec,用 export_model 导出(固定输入 [3,32,320]),再用 paddle_lite_opt 转 .nb,替换进官方 Android demo / 自有 App

现象:

  • 使用官方提供的 ch_PP-OCRv2_rec_infer_opt.nb 时,App 正常运行。
  • 使用自训练权重「导出 → opt」得到的 .nb 替换后,运行报错:Check failed: (k_ == w_dims[0]): 9088!==128(或类似维度不匹配)。
  • 说明当前公开的 ch_PP-OCRv2_rec_infer 推理模型经同一套 opt 得到的图与官方发布的 .nb 图结构不一致(FC 输入维 128 vs 实际 9088),导致自训练模型无法按文档描述“替换模型”在端上使用。

需求:

  • 为在 Android 端内集成自训练识别模型,请求提供以下之一:
    1)与「ch_PP-OCRv2_rec_infer_opt.nb」图结构完全一致的 pdmodel(或 pdmodel+pdiparams),或
    2)当年生成该 .nb 时使用的导出配置(含固定 input shape)和 Paddle-Lite opt 命令/脚本。

这样我们可以用同套图 + 自训练权重重新 opt 得到可在当前 App 内直接替换的 .nb,与文档中“可替换模型”的说明一致。谢谢。

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions