Class NNModelSelection
NNModelSelection Class
Description
A class used to select Neural Network (NN) task and models.
Syntax
class NNModelSelection
Members
Public Constructors |
|
---|---|
NNModelSelection::NNModelSelection |
Constructs a NNModelSelection object. |
Public Methods |
|
NNModelSelection::modelSelect |
Select NN task and models. |
NNModelSelection::modelSelect
Description
Select Neural Network (NN) task and models.
Syntax
void modelSelect(unsigned char nntask);
void modelSelect(unsigned char nntask, unsigned char objdetmodel, unsigned char facedetmodel, unsigned char facerecogmodel);
Parameters
nntask: Neural network task to perform. Default value is NA_MODEL.
OBJECT_DETECTION, FACE_DETECTION, FACE_RECOGNITION
objdetmodel: Neural network model used for Object Detection. Default value is NA_MODEL.
DEFAULT_YOLOV3TINY, CUSTOMIZED_YOLOV3TINY (YOLOv3 model)
DEFAULT_YOLOV4TINY, CUSTOMIZED_YOLOV4TINY (YOLOv4 model)
DEFAULT_YOLOV7TINY, CUSTOMIZED_YOLOV7TINY (YOLOv7 model)
facedetmodel: Neural network model used for Face Detection. Default value is NA_MODEL.
DEFAULT_SCRFD, CUSTOMIZED_SCRFD
facerecogmodel: Neural network model used for Face Recognition. Default value is NA_MODEL.
DEFAULT_MOBILEFACENET, CUSTOMIZED_MOBILEFACENET
Returns
NA
Example Code
Example: ObjectDetectionCallback
Note
“NNModelSelection.h” must be included to use the class function. Replace objdetmodel, facedetmodel and facerecogmodel arguments with “NA_MODEL” if they are not necessary for your selected NN Task.