1 Star 0 Fork 35

CpDec/FastDeploy

forked from PaddlePaddle/FastDeploy 
加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
文件
克隆/下载
贡献代码
同步代码
取消
提示: 由于 Git 不支持空文件夾,创建文件夹后会生成空的 .keep 文件
Loading...
README

SCRFD RKNPU2部署模型

下载预训练ONNX模型

为了方便开发者的测试,下面提供了SCRFD导出的各系列模型,开发者可直接下载使用。(下表中模型的精度来源于源官方库)

模型 大小 精度
SCRFD-500M-kps-160 2.5MB -
SCRFD-500M-160 2.2MB -
SCRFD-500M-kps-320 2.5MB -
SCRFD-500M-320 2.2MB -
SCRFD-500M-kps-640 2.5MB 90.97%
SCRFD-500M-640 2.2MB 90.57%
SCRFD-1G-160 2.5MB -
SCRFD-1G-320 2.5MB -
SCRFD-1G-640 2.5MB 92.38%
SCRFD-2.5G-kps-160 3.2MB -
SCRFD-2.5G-160 2.6MB -
SCRFD-2.5G-kps-320 3.2MB -
SCRFD-2.5G-320 2.6MB -
SCRFD-2.5G-kps-640 3.2MB 93.8%
SCRFD-2.5G-640 2.6MB 93.78%
SCRFD-10G-kps-320 17MB -
SCRFD-10G-320 15MB -
SCRFD-10G-kps-640 17MB 95.4%
SCRFD-10G-640 15MB 95.16%
SCRFD-10G-kps-1280 17MB -
SCRFD-10G-1280 15MB -

导出ONNX模型

#下载scrfd模型文件
e.g. download from  https://onedrive.live.com/?authkey=%21ABbFJx2JMhNjhNA&id=4A83B6B633B029CC%215542&cid=4A83B6B633B029CC

# 安装官方库配置环境,此版本导出环境为:
- 手动配置环境
  torch==1.8.0
  mmcv==1.3.5
  mmdet==2.7.0

- 通过docker配置
  docker pull qyjdefdocker/onnx-scrfd-converter:v0.3

# 导出onnx格式文件
- 手动生成
  python tools/scrfd2onnx.py configs/scrfd/scrfd_500m.py weights/scrfd_500m.pth --shape 640 --input-img face-xxx.jpg

- docker
  docker的onnx目录中已有生成好的onnx文件

ONNX模型转换RKNN模型

wget  https://bj.bcebos.com/paddlehub/fastdeploy/scrfd_500m_bnkps_shape640x640.onnx
python tools/rknpu2/export.py --config_path tools/rknpu2/config/RK3588/scrfd.yaml

详细部署文档

版本说明

马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
Python
1
https://gitee.com/martinfields/FastDeploy.git
git@gitee.com:martinfields/FastDeploy.git
martinfields
FastDeploy
FastDeploy
develop

搜索帮助