#下载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文件
为了方便开发者的测试,下面提供了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 | - |
- 本版本文档和代码基于SCRFD CommitID:17cdeab 编写