求助:S100 是否有可用的人脸检测 HBM 模型或转换示例?

大家好,我在 S100 板子上做人脸跟踪项目,想把人脸检测跑到 BPU 上。

目前板子自带的模型目录里主要是通用检测模型,例如 yolo11n_detect_nashe_640x640_nv12.hbm、yolov8_640x640_nv12.hbm 等,但这些不是专门的人脸检测模型,实际用于云台追脸时容易出现目标不准或误检。

我现在尝试把 yolov8-face.pt 转成 HBM,流程已经跑通到:
pt → onnx → hbm
HBM 在板端可以加载,也能 forward,输出为:
output0 [1, 5, 2100]

但目前遇到的问题是:HBM 输出的 score 分支异常,NV12/校准版本测试中出现 score 全 0 或 score 固定不变的情况。ONNX 在 PC 端使用 Ultralytics 推理是正常的,可以检测到人脸。

想请教一下:

  1. 社区里是否有已经转换好的 S100/S600 可用人脸检测 HBM?
    例如 SCRFD、RetinaFace、YuNet、yolov8-face 等。

  2. 是否有 S100 上人脸检测模型的 hb_compile 配置示例?
    特别是 input_type_rt/input_type_train、calibration 数据格式、quant_config 相关配置。

  3. 如果使用 yolov8-face 这类输出为 [1,5,2100] 的模型,是否需要对输出层或 score 分支做特殊处理?

我的工具链版本:
S100/S600 ai_toolchain v3.7.0
hb_compile 3.5.3
hbdk 4.7.5
hmct 2.6.5

板端环境:
DNN 3.13.6
BPU Lib 2.1.2

感谢大家,如果有模型、配置文件或 demo 可以参考,非常感谢。

没事了问题已经解决了参考万字长文,学妹一看就会的RDKS100模型量化及部署csdn上面的资料

你好,人脸比较敏感,一般不会提供这一类方案

请问你是通过pt转的吗?还是使用的已有的模型,config的配置参数可以分享一个 吗