AMB82 SCRFD Convert To ONNX

SCRFD

SCRFD 是一种最初在 Arxiv 中进行了描述,高效的高精度人脸检测方法。
它提供了一个易于使用的管道来训练具有 NAS 支持的高效人脸检测器。
你可以拜访 [InsignFace](https://github.com/deepinsight/insightface/tree/master/detection/scrfd) 获取更多信息。
对于amebaPro2边缘计算,建议makers使用 [SCRFD_500M_KPS](https://1drv.ms/u/s!AswpsDO2toNKri_NDM0GIkPpkE2f?e=JkebJo).

SCRFD 设置环境

SCRFD默认格式是pytorch,所以要转成ONNX格式:

cd ~/insightface/detection/scrfd
python tools/scrfd2onnx.py configs/scrfd/scrfd_500m_bnkps.py scrfd_500mkps.pt --shape 640 640

你可以在 [InsignFace] 中找到 tools/scrfd2onnx.py(https://github.com/deepinsight/insightface/tree/master/detection/scrfd).

如果Makers遇到Python包依赖问题,可以尝试用virtualenv创建隔离的Python环境,参考我们的requirements_scrfd.txt安装Python包。

pip install -r requirements_scrfd.txt

为了向您展示如何快速翻译和导出 SCRFD,我们可以从网上下载预训练的权重:
[scrfd.lite.ai.toolkit](https://github.com/DefTruth/scrfd.lite.ai.toolkit)
请下载`scrfd_500m_bnkps_shape640x640.onnx`或`scrfd_500m_bnkps_shape320x320.onnx`用于amebaPro2边缘计算。

下载

请先确认已安装QQ通讯软体