MobileNetV3-based real-time human head classification detection method

Abstract

The invention provides a MobileNetV3-based real-time human head classification detection method. The method comprises the following steps of S1, selecting a training data set and performing classification labeling on the selected training data set, wherein the training data set comprises a mask face data set, a face data set and a head data set; S2, performing enhancement processing on the image data of the labeled training data set; S3, generating a priori box for each category according to different categories of the training data set; S4, constructing a model based on the MobileNetV3, and training the model through the training data set; and S5, applying the trained model to the large screen, and reasoning the real-time human head image captured by the large screen camera to obtain a classification detection result. According to the invention, rapid and accurate classification detection of the human head is realized, key information points of the human face are provided, and a classification detection result provides a data basis for subsequent processing.

Description

technical field [0001] The invention belongs to the technical field of face recognition, and in particular relates to a real-time head classification and detection method based on MobileNetV3. Background technique [0002] MobileNetV3, a small and efficient network structure proposed by Google for mobile or embedded device applications. [0003] FPN, short for Feature Pyramid Net, feature pyramid network. [0004] SSH, the abbreviation of Single Stage Headless Face Detector, single-step face recognition. [0005] At present, the accuracy and speed of face recognition technology can meet the requirements of the application, and the tracking and statistics products based on face recognition have been implemented. However, due to the impact of the epidemic, the following problems have emerged: [0006] (1) Wearing a mask makes the tracking statistics based on face recognition wrong; [0007] (2) The statistics based on head detection generally only count the flow of people, ...

AI Technical Summary

Problems solved by technology

[0013] Aiming at the above-mentioned defects of traditional face recognition tracking due to the influence of wearing a mask in the prior art, and directly adding head detection to the existing face recognition, the defects of high cost, low precision and no classification, the present invention provides a real-time head detection based on MobileNetV3. Classification detection method to solve the above technical problems