基于OpenVINO部署PaddlePadle-YOLOE模型—2.模型下载与转换

原创

已于 2022-10-28 17:57:49 修改 · 1.6k 阅读

6 ·

CC 4.0 BY-SA版权

文章标签：

#openvino #paddlepaddle

于 2022-10-28 17:47:09 首次发布

本文介绍了PP - YOLOE模型在OpenVINO中的转换流程。先下载PP - YOLOE官方训练模型，接着对直接导出无法在OpenVINO使用的模型进行裁剪，去掉后处理过程。然后将裁剪后的模型转为ONNX格式并指定bath_size大小，最后利用OpenVINOTM自带工具将其转为IR格式以提升推理速度。

1. 模型下载

首先下载PP-YOLOE官方训练模型，该模型由PaddleDetection提供，基于COCO数据集训练，可以识别80种常见物体。此处采用的是PaddleDetection release/2.5版本，PP-YOLOE+模型，具体可以参考官方文件PP-YOLOE。

使用命令，导出我们要使用的模型，在命令行种依次输入以下指令，导出我们所使用的模型文件：

// 打开PaddleDetection代码文件
cd ./PaddleDetection 
// 导出指定模型
python tools/export_model.py -c configs/ppyoloe/ppyoloe_plus_crn_l_80e_coco.yml -o weights=https://paddledet.bj.bcebos.com/models/ppyoloe_plus_crn_l_80e_coco.pdparams

此处导出的是PP-YOLOE+模型，l_80e格式，导出命令输出如下图所示。

在这里插入图片描述