Class NNModelSelection

NNModelSelection 类

说明

用于选择神经网络(NN)任务和模型的类。

语法

class NNModelSelection

成员

公共构造函数

NNModelSelection::NNModelSelection

构造一个 NNModelSelection 对象。

公共方法

NNModelSelection::modelSelect

选择 NN 任务和模型。

NNModelSelection::modelSelect

说明

选择神经网络(NN)任务和模型。

语法

void modelSelect(unsigned char nntask);
void modelSelect(unsigned char nntask, unsigned char objdetmodel, unsigned char facedetmodel, unsigned char facerecogmodel);

参数

nntask: 要执行的神经网络任务。默认值为 NA_MODEL。

  • OBJECT_DETECTION、FACE_DETECTION、FACE_RECOGNITION

objdetmodel: 用于目标检测的神经网络模型。默认值为 NA_MODEL。

  • DEFAULT_YOLOV3TINY、CUSTOMIZED_YOLOV3TINY(YOLOv3 模型)

  • DEFAULT_YOLOV4TINY、CUSTOMIZED_YOLOV4TINY(YOLOv4 模型)

  • DEFAULT_YOLOV7TINY、CUSTOMIZED_YOLOV7TINY(YOLOv7 模型)

facedetmodel: 用于人脸检测的神经网络模型。默认值为 NA_MODEL。

  • DEFAULT_SCRFD、CUSTOMIZED_SCRFD

facerecogmodel: 用于人脸识别的神经网络模型。默认值为 NA_MODEL。

  • DEFAULT_MOBILEFACENET、CUSTOMIZED_MOBILEFACENET

返回值

NA

示例代码

示例:ObjectDetectionCallback

备注

必须包含"NNModelSelection.h"才能使用该类函数。如果所选 NN 任务不需要 objdetmodel、facedetmodel 和 facerecogmodel 参数,请将其替换为"NA_MODEL"。