大家好,我在 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 推理是正常的,可以检测到人脸。
想请教一下:
-
社区里是否有已经转换好的 S100/S600 可用人脸检测 HBM?
例如 SCRFD、RetinaFace、YuNet、yolov8-face 等。 -
是否有 S100 上人脸检测模型的 hb_compile 配置示例?
特别是 input_type_rt/input_type_train、calibration 数据格式、quant_config 相关配置。 -
如果使用 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 可以参考,非常感谢。