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
示例代码
备注
必须包含"NNModelSelection.h"才能使用该类函数。如果所选 NN 任务不需要 objdetmodel、facedetmodel 和 facerecogmodel 参数,请将其替换为"NA_MODEL"。