人脸检测

在一张图片中找出人脸, 并且框出人脸,即知道脸的位置和大小

使用了YOLO V2模型对人脸进行检测

1. 使用方法:

  • 下载模型: 到这里 下载 face_model_at_0x300000.kfpkg 这个模型文件
  • 用 kflash_gui 下载模型到 Flash, 或者放到 SD 卡中
  • 加载模型
    task = kpu.load(0x300000)
    # task = kpu.load("/sd/face.kmodel")
    anchor = (1.889, 2.5245, 2.9465, 3.94056, 3.99987, 5.3658, 5.155437, 6.92275, 6.718375, 9.01025)
    kpu.init_yolo2(task, 0.5, 0.3, 5, anchor)
    
    因为使用了YOLO V2这个模型, 它有专用的函数接口, 使用init_yolo2来初始化模型

参数分别为:

  • kpu_net: kpu 网络对象, 即加载的模型对象, KPU.load()的返回值
  • threshold: 概率阈值, 只有是这个物体的概率大于这个值才会输出结果, 取值范围:[0, 1]
  • nms_value: box_iou 门限, 为了防止同一个物体被框出多个框,当在同一个物体上框出了两个框,这两个框的交叉区域占两个框总占用面积的比例 如果小于这个值时, 就取其中概率最大的一个框
  • anchor_num: anchor 的锚点数, 这里固定为 len(anchors)//2
  • anchor: 锚点参数与模型参数一致,同一个模型这个参数是固定的,和模型绑定的(训练模型时即确定了), 不能改成其它值。

然后输入图片数据,运行模型

code = kpu.run_yolo2(task, img)

得到结果, 完整例程看这里

API 文档看 Maix.KPU

powered by GitbookFile Modify: 2021-01-16 11:36:59

results matching ""

    No results matching ""