转换PP-OCRV5_det模型的时候遇到报错：AssertionError: when model typs is det, backbone only support ['MobileNetV3', 'ResNet', 'ResNet_vd', 'ResNet_SAST', 'PPLCNet', 'PPLCNetV3', 'PPHGNet_small', 'PPLCNetV2_base', 'RepSVTR_det']

[jamir665]

问题描述

如题，我想将模型转换为.pdmodel文件来使用，但加载了这个 yml文件：https://github.com/PaddlePaddle/PaddleOCR/blob/main/configs/det/PP-OCRv5/PP-OCRv5_server_det.yml

然后执行了这样的转换指令：

#!/bin/bash

# 需要注意一下，paddlepaddle在3.0.0版本已经不会导出.pdmodel文件了，如果需要导出.pdmodel文件，需要使用paddlepaddle2.6.1版本

# 切换到项目根目录
cd $(dirname "$0")

# 激活 PaddlePaddle 环境

# conda init

# conda activate paddlepaddle_env

# 定义配置文件和模型权重路径
CONFIG_FILE="model/OCR/ch-OCR/det/PP-OCRv5_server_det_infer/config.yml"
MODEL_CHECKPOINT="model/OCR/ch-OCR/det/PP-OCRv5_server_det_infer/inference"
OUTPUT_DIR="./output/ch-OCR/det/PP-OCRv5_inference"

# 运行模型导出脚本
python tools/export_model.py --config ${CONFIG_FILE} --opt Global.checkpoints=${MODEL_CHECKPOINT} Global.save_inference_dir=${OUTPUT_DIR}

echo "模型转换完成！导出的模型在 ${OUTPUT_DIR} 目录下。" 

之后会提示报错：

似乎是backbone不支持导致的，但这个yml文件不就是官方适配最新的V5模型的吗？

paddlepaddle版本：