HUMAN HEAD DETECTION METHOD, ELETRONIC DEVICE AND STORAGE MEDIUM
A method for detecting and tracking human head in an image by an electronic device is disclosed. The method may include segmenting the image into one or more sub-images; inputting each sub-image to a convolutional neural network trained according to training images having marked human head positions; outputting by a preprocessing layer of the convolutional neural network comprising a first convolutional layer and a pooling layer, a first feature corresponding to each sub-image; mapping through a second convolutional layer the first feature corresponding to each sub-image to a second feature corresponding to each sub-image; mapping through a regression layer the second feature corresponding to each sub-image to a human head position corresponding to each sub-image and a corresponding confidence level of the human head position; and filtering, according to the corresponding confidence level, human head positions corresponding to the one or more sub-images, to acquire detected human head positions in the image.
Latest Tencent Technology (Shenzhen) Company Limited Patents:
- Restoring a video for improved watermark detection
- Data processing method, device, and storage medium
- Speech recognition method and apparatus, device, storage medium, and program product
- Picture encryption method and apparatus, computer device, storage medium and program product
- Video decoding and encoding method and apparatus, computer-readable storage medium and electronic device
This application is a continuation application of the International PCT Application No. PCT/CN2018/070008, filed with the Chinese Patent Office on Jan. 2, 2018 and claims priority to Chinese Patent Application No. 2017100292446, filed with the Chinese Patent Office on Jan. 16, 2017 and entitled “HUMAN HEAD DETECTION METHOD AND APPARATUS”, which is incorporated herein by reference in its entirety.
FIELD OF THE TECHNOLOGYThis application relates to the technical field of image processing, and in particular, to a method, an electronic device and a storage medium for human head detection.
BACKGROUND OF THE DISCLOSUREHuman head detection refers to the detection of the head of a human body in an image, and a result of the human head detection has various applications, such as applications in the field of security. At present, the human head detection is implemented mainly based on a shape and color of a human head. At present, a specific process of the human head detection includes: first, binarizing image pixels, and then performing edge detection to acquire a substantially circular edge; and then using circle detection to acquire a position and size of the circular edge, and then performing gray scale and size determination on a corresponding circular area in the original image to obtain human head detection.
However, currently, the human head detection relies on an assumption that the shape of the human head is circular. In fact, the shape of the human head is not strictly circular, and the shapes of the human heads of different person are also different. As a result, during the current human head detection, some human heads miss the detection and accuracy of the result of the human head detection is relatively low.
SUMMARYAccording to various embodiments provided by this disclosure, methods, an electronic devices and a storage media are provided for implementing human head detection in images.
A human head detection method includes:
segmenting, by an electronic device, an image to be detected into one or more sub-images;
inputting, by the electronic device, each sub-image to a convolutional neural network trained according to a training image having a marked human head position respectively, and outputting, by a preprocessing layer including a convolutional layer and a pooling layer in the convolutional neural network, a first feature corresponding to each sub-image;
mapping, by the electronic device through a convolutional layer after the preprocessing layer in the convolutional neural network, the first feature corresponding to each sub-image to a second feature corresponding to each sub-image;
mapping, by the electronic device through a regression layer in the convolutional neural network, the second feature corresponding to each sub-image to a human head position corresponding to each sub-image and a corresponding confidence level of the human head position; and
filtering, by the electronic device according to the corresponding confidence level, the human head position corresponding to each sub-image, to acquire a human head position detected in the image to be detected.
An electronic device includes a memory and a processor, the memory storing a computer readable instruction, the computer readable instruction, when executed by the processor, causing the processor to perform the following steps:
segmenting an image to be detected into one or more sub-images;
inputting each sub-image to a convolutional neural network trained according to a training image having a marked human head position respectively, and outputting, by a preprocessing layer including a convolutional layer and a pooling layer in the convolutional neural network, a first feature corresponding to each sub-image;
mapping, through a convolutional layer after the preprocessing layer in the convolutional neural network, the first feature corresponding to each sub-image to a second feature corresponding to each sub-image;
mapping, through a regression layer of the convolutional neural network, the second feature corresponding to each sub-image to a human head position corresponding to each sub-image and a corresponding confidence level of the human head position; and
filtering, according to the corresponding confidence level, the human head position corresponding to each sub-image, to acquire the human head position detected in the image to be detected.
One or more non-volatile storage media storing a computer readable instruction is provided, the computer readable instruction, when executed by one or more processors, causing the processor to perform the following steps:
segmenting an image to be detected into one or more sub-images;
inputting each sub-image to a convolutional neural network trained according to a training image having a marked human head position respectively, and outputting, by a preprocessing layer including a convolutional layer and a pooling layer in the convolutional neural network, a first feature corresponding to each sub-image;
mapping, through a convolutional layer after the preprocessing layer in the convolutional neural network, the first feature corresponding to each sub-image to a second feature corresponding to each sub-image;
mapping, through a regression layer in the convolutional neural network, the second feature corresponding to each sub-image to a human head position corresponding to each sub-image and a corresponding confidence level of the human head position; and
filtering, according to the corresponding confidence level, the human head position corresponding to each sub-image, to acquire the human head position detected in the image to be detected.
Details of one or more embodiments of this application are put forward in the following accompanying drawings and descriptions. Other features, objectives, and advantages of this application become more obvious with reference to the specification, the accompanying drawings, and the claims.
To describe the technical solutions of the embodiments of this application more clearly, the following briefly introduces the accompanying drawings required for describing the embodiments. The accompanying drawings described below are only some embodiments of this application, and a person of ordinary skill in the art can obtain other accompanying drawings according to these accompanying drawings without creative efforts.
To make the objectives, technical solutions, and advantages of this application clearer, the following disclosure further describes this application in detail with reference to the accompanying drawings and embodiments. It should be understood that specific embodiments described herein are merely intended to explain this application and are not intended to limit this application.
While the disclosure herein specifically refer to human head detection in top view images, the underlying principle may be applied to detection of other objects in any type of images. For example, the systems and methods disclosed below may be applied to detection of motor vehicles in satellite images for monitoring traffic and the like.
In an embodiment, the electronic device 110 may be configured to segment an image to be detected into one or more sub-images; input each sub-image to a convolutional neural network trained according to training images having marked human head positions (or labeled with human heads)y, and output, by a preprocessing layer including at least one convolutional layer and at least one pooling layer in the convolutional neural network, a first feature corresponding to each sub-image; map, through at least one another convolutional layer after the preprocessing layer in the convolutional neural network, the first feature corresponding to each sub-image to a second feature corresponding to each sub-image; map, through at least one regression layer of the convolutional neural network, the second feature corresponding to each sub-image to a human head position corresponding to each sub-image and a corresponding confidence level of the human head position; and filter, according to the corresponding confidence level, the human head position corresponding to each sub-image, to acquire a human head positions detected in the image to be detected.
S302: Segment an image to be detected into one or more sub-images.
The image to be detected is an image on which human head detection needs to be performed. The image to be detected may be a picture or a video frame in a video. The sub-images are images which are segmented from the image to be detected and have a size smaller than the image to be detected. All segmented sub-images may have the same size or different sizes.
Specifically, the electronic device may traverse a window of a fixed size in the image to be detected according to a transverse step size and a longitudinal step size, thereby segmenting the sub-images having the same size as the window size from the image to be detected during the traversal process. The segmented sub-images may be combined to the image to be detected.
In an embodiment, step S302 includes: segmenting the image to be detected to one or more sub-images of a fixed size, adjacent sub-images in the segmented sub-images having an overlapping part.
The adjacent sub-images refer to that positions of the sub-images in the image to be detected are adjacent, and the adjacent sub-images may partially overlap. Specifically, the electronic device may traverse the window of a fixed size in the image to be detected according to the transverse step size smaller than a window width and the longitudinal step size smaller than a window height, to acquire one or more sub-images of the same size, and adjacent sub-images have an overlapping part.
In this embodiment, there is an overlapping part between the segmented adjacent sub-images, thereby ensuring that the adjacent sub-images have higher correlation, and improving accuracy of detecting a human head position from the image to be detected, particularly when a human head lies at boundary between the adjacent sub-images.
S304: Input each sub-image to a convolutional neural network trained according to a set of training images having marked human head positions, and output, by a preprocessing layer including at least one convolutional layer and at least one pooling layer in the convolutional neural network, a first feature corresponding to each sub-image.
The Convolutional Neural Network (CNN) is an artificial intelligence neural network. The convolutional neural network includes a preprocessing layer having at least one convolutional layer and at least one pooling layer. The convolutional neural network used in this embodiment may be directly constructed, and may alternatively be acquired by reconstructing an existing convolutional neural network. A computational task in the convolutional neural network may be implemented by a central processing unit or a graphics processing unit. Time consumed by the central processing unit for human head detection is proximately a level of seconds, and time consumed by the graphics processing unit for human head detection may be reduced to a level of hundred milliseconds, thereby realizing real-time human head detection.
In the convolutional layer in the convolutional neural network, there are a plurality of feature maps, each feature map includes a plurality of neurons, and all neurons of the same feature map share one convolution kernel. The convolution kernel provides a weight of the corresponding neuron, and the convolution kernel represents a feature. The convolution kernel is generally initialized in a form of a random decimal matrix, and a proper convolution kernel will be learned during training of the network to represent a feature. The convolutional layer can reduce connection between various layers in the neural network, and in addition, a risk of overfitting is reduced.
Pooling may take two exemplary forms of implementation: mean pooling and max pooling. Pooling may be considered as a special convolutional process. Convolution and pooling greatly simplify complexity of the neural network and reduce parameters of the neural network.
The training images having human heads therein may be pre-marked (or labeled) with human head positions For example, human head positions in the training images may be manually marked or labeled, or may be marked or labeled using other automatic means. The training images having the marked human head positions and the image to be detected may be images captured in the similar scene, setting or background, thereby further improving the accuracy of human head detection. The training image having marked human head positions can be of the same size or different sizes as the image to be detected.
In an embodiment, when the convolutional neural network is trained, a confidence level may be assigned to the human head position marked in the training image. The training image is segmented into one or more sub-images according to the same segmentation manner as that of the image to be detected. The segmented sub-images are separately input to the convolutional neural network, and the convolutional neural network outputs human head positions and confidence levels. A difference between the output head positions and the marked head position is calculated, and a difference between the corresponding confidence levels is calculated. According to the two differences, parameters of the convolutional neural network are adjusted. The training is continued until a termination condition is reached. The termination condition may be that each difference is less than a preset difference threshold, or the number of iterations reaches a preset number of times.
The preprocessing layer is used above as a general term of other layers in the convolutional neural network except for the regression layer and a convolutional layer before the regression layer. The preprocessing layer may include at least one convolutional layer and at least one pooling layer. The preprocessing layer may include parallel convolutional layers, and data output by the parallel convolutional layers may be spliced and input to a next layer. The last layer in the preprocessing layer may be a convolutional layer or a pooling layer. The preprocessing layer may include multiple pairs of convolutional layer and pooling layer connected in tandem. The preprocessing layer may include additional rectifying layers.
S306: Map, through a convolutional layer after the preprocessing layer in the convolutional neural network, a first feature corresponding to each sub-image to a second feature corresponding to each sub-image.
A conventional convolutional neural network is generally used for classification, and the preprocessing layer in the convolutional neural network for classification is followed by a fully connected layer. The fully connected layer may map the first feature output by the preprocessing layer to probability data corresponding to each preset type (or class). Therefore, a type to which an input image belongs may be determined by the regression layer. In this embodiment, the convolutional neural network is used for human head detection rather than classification. As such, the convolutional layer is configured to replace the fully connected layer, and to output the second feature for describing the sub-image features. The number of the second features corresponding to each sub-image may be plural.
S308: Map, through a regression layer in the convolutional neural network, the second feature corresponding to each sub-image to a human head position corresponding to each sub-image and a confidence level corresponding to the human head position.
The human head position may be represented by a position of a rectangular box bounding a human head in the image. The position of the rectangular box may be represented by a quadruple. The quadruple may include a horizontal coordinate and a longitudinal coordinate of one vertex of the rectangular box and a width and a height of the rectangular box. Alternatively, the quadruple may include a horizontal coordinate and a longitudinal coordinate of each of two diagonal vertexes of the rectangular box. The confidence level output by the regression layer are in a one-to-one correspondence with the human head position output by the regression layer, thereby indicating a probability that the corresponding rectangular box does correspond to a human head at the corresponding position in the image. The regression layer may use a support vector machine (SVM).
In an embodiment, step S308 includes: mapping, through the convolutional layer in the regression layer in the convolutional neural network, the second feature corresponding to each sub-image to the human head position corresponding to each sub-image and the confidence level corresponding to the human head position. Specifically, the electronic device may directly map the second feature corresponding to each sub-image to the human head position corresponding to each sub-image and the confidence level corresponding to the human head position through the same convolutional layer in the regression layer in the convolutional neural network.
In an embodiment, step S308 includes: mapping, through a first convolutional layer in the regression layer in the convolutional neural network, the second feature corresponding to each sub-image to the human head position corresponding to each sub-image; and mapping, through a second convolutional layer in the regression layer in the convolutional neural network, the second feature corresponding to each sub-image to the confidence level corresponding to the output human head position.
For example, referring to
S310: Filter, according to the corresponding confidence level, the human head position corresponding to each sub-image, and acquire a human head position detected in the image to be detected.
Specifically, the electronic device may compare the confidence level of each human head position output by the convolutional neural network with a confidence level threshold, and filter out human head positions of which confidence levels are less than the confidence level threshold. The electronic device may further filter the human head positions, of which areas are smaller than a preset area, in the human head positions filtered by using the confidence level threshold. The electronic device may cluster the filtered human head positions to combine the plurality of human head positions of the same type to acquire one combined human head position in the image to be detected, or select one of the plurality of human head positions clustered to the same type as the human head position in the image to be detected.
According to the foregoing human head detection method, the convolutional neural network is trained in advance based on the training images having the marked human head position, and the convolutional neural network can automatically learn human head features. The trained convolutional neural network can automatically extract appropriate features from the sub-images to output candidate human head positions and corresponding confidence levels, and then filter, according to the confidence levels, to acquire the human head position in the image to be detected. The human head shape is learned rather than pre-assumed. As such a missed detection caused by presuming the shape of the human head can be avoided, and accuracy of the human head detection is improved. Moreover, in the convolutional neural network, the first features of the sub-images are output by the preprocessing layer including the convolutional layer and the pooling layer, and the second features are outputted by the convolutional layer after the preprocessing layer and before the regression layer to accurately describe human head features in the sub-images. Therefore, the second features are directly mapped to the human head positions and confidence levels by the regression layer, which is new application of the convolutional neural network of the new structure. Compared with the traditional circle detection, the accuracy of the human head detection is greatly improved.
In an embodiment, before step S302, the human head detection method further includes a step of converting and training the convolutional neural network for classification to a convolutional neural network for human head detection. Referring to
S502: Convert a fully connected layer after the preprocessing layer and before the regression layer included in the convolutional neural network for classification to a convolutional layer.
A conventional convolutional neural network for classification is a trained convolutional neural network which can classify images input to the convolutional neural network, such as GoogleNet, VGGNET, or AlexNet. The convolutional neural network for classification includes the preprocessing layer, the fully connected layer, and the regression layer. The fully connected layer is configured to output second features corresponding to each preset type (or class) of the conventional classification application.
The sparse connection and weight sharing of the fully connected layer and the convolutional layer are different. Each neuron of the fully connected layer is connected to all neurons of a preceding layer. Both the convolutional layer and the fully connected layer acquire input of a next layer by multiplying output of the preceding layer by a parameter matrix. As such, the conventional fully connected layer can be converted to the convolutional layer by changing an arrangement manner of parameters of the fully connected layer.
S504: Replace the regression layer in the convolutional neural network for classification with a regression layer configured to map the second feature output by the converted convolutional layer to the human head position and the corresponding confidence level.
In the conventional convolutional neural network for classification, the regression layer is configured to map the second features of each preset type output by the fully connected layer to a probability corresponding to each preset type, and determine, according to the mapped probability, a preset type to which the image belongs. For example, a preset type corresponding to a maximum probability is selected as the preset type to which the input image belongs.
In the convolutional neural network for human head detection of this disclosure, the regression layer is configured to map a preset number of second features output by the converted convolutional layer to the human head positions and the confidence levels corresponding to the human head positions. The regression layer may use a convolutional layer. The convolutional layer directly maps the second features to the human head positions and the confidence levels corresponding to the human head positions. The regression layer may also use two convolutional layers in parallel. One convolutional layer is configured to map the second features to the human head positions, and the other convolutional layer is configured to map the second features to the confidence levels corresponding to the mapped human head positions.
S506: Train the convolutional neural network including the preprocessing layer, the converted convolutional layer, and the replaced regression layer by using the training images having the marked human head positions.
The convolutional neural network including the preprocessing layer, the converted convolutional layer and the replaced regression layer is reconstructed and modified from the conventional convolutional neural network for classification applications. In one implementation, parameters of the preprocessing layer may be pre-trained. Then for the reconstructed convolutional neural network, mainly the parameters in the converted convolutional layer and the replaced regression layer need to be trained. The training processing may be joint process. For example, the entire network may be trained. The preprocessing layer training parameters may be initialized to its pre-trained parameters and retrained together with the rest of network.
Specifically, when the reconstructed convolutional neural network is trained, the confidence level may be pre-assigned to the marked human head positions of the training image. The training image is segmented into one or more sub-images according to the same segmenting manner as that of the image to be detected. The segmented sub-images are respectively input to the convolutional neural network, and the human head positions and the confidence levels are output by the preprocessing layer, the convolutional layer after the preprocessing layer, and the regression layer of the convolutional neural network. The difference between the output human head positions and the marked human head position is calculated, and the difference between the corresponding confidence levels is calculated, and the parameters in the preprocessing layer, the convolutional layer after the preprocessing layer, and the regression layer in the convolutional neural network are adjusted according to the two differences. The training is continued until a termination condition is reached. The termination condition may be that the difference is less than a preset difference, or the number of iterations reaches a preset number of times.
In this embodiment, the training is performed after reconstruction of conventional neural network for classification into a the convolutional neural network for human head detection. The reconstruction does not require complete redesign of the neural network, training duration can be reduced and efficiency of human head detection is improved.
As shown in
S602: Screen, from the human head positions corresponding to the sub-images, to acquire a human head position corresponding to a confidence level greater than or equal to a confidence level threshold.
Specifically, the electronic device may form the human head positions respectively corresponding to the sub-images segmented from the image to be detected into a human head position set, traverse the human head position set, and compare confidence levels the traversed human head positions with the confidence level threshold. The human head positions having confidence levels lower than the confidence level threshold may be removed from the human head position set. The remaining human head positions in the human head position set after the traversing are the acquired human head positions of which the corresponding confidence levels are greater than or equal to the confidence level threshold. The confidence level threshold may be set as needed, for example, may be valued from 0.5 to 0.99.
S604: Selecting or identifying human head positions among the screened human head positions in S602 that intersect in the image to be detected.
The intersection of the human head positions means that enclosed areas indicated by respective human head positions have an intersection in the image to be detected. When the human head position is represented by a position of a rectangular box including the human head image, the intersection of the human head positions is the intersection of the corresponding rectangular boxes. Specifically, the electronic device may select a human head position intersecting with the acquired human head position in the image to be detected from the human head position set formed by the human head positions respectively corresponding to all the sub-images segmented from the image to be detected. The electronic device may also seek for the intersecting human head positions from only the acquired human head positions.
S606: Determine, according to the acquired human head position and the identified human head position, the human head position detected in the image to be detected.
Specifically, the electronic device may classify the acquired human head positions and the selected human head positions. Each type includes at least one of the acquired human head positions, and also includes human head positions intersecting with the at least one human head position. The electronic device may combine the human head positions of each type to one human head position as a detected head position, or select one human head position from the human head positions of each type as the detected human head position.
In this embodiment, the accuracy of human head detection can be further improved by using the confidence levels and the position intersection the basis for determining the human head position in the image to be detected.
As shown in
S702: Use the acquired human head position (from step 602) and the selected human head position (from step 604) as nodes in a bipartite graph, as a first group and second group, respectively.
The bipartite graph is a graph in the graph theory, the nodes in the bipartite graph may be segmented into two groups, and all edges connected to the nodes are caused to span boundaries of the groups.
S704: Assign default and positive weights to the edges between the nodes in the bipartite graph.
There is an edge between each acquired human head position and the correspondingly selected intersecting human head position. The default and positive weight is a positive value, such as 1000.
S706: Reduce the correspondingly assigned weights for the corresponding edges that are associated with nodes representing intersecting head positions.
Specifically, when the human head positions indicated by the nodes associated with the edges intersect, the electronic device may subtract a positive value less than the default and positive weight from the correspondingly assigned weight, and then divide the subtracted value by the default and positive weight to acquire an updated weight. If the default and positive weight is 1000, and the positive value less than the default and positive weight is 100, then the updated weight is (1000−100)/1000=0.9.
S708: Solve a maximum weight edge combination of the bipartite graph, and acquire the human head position detected in the image to be detected.
An edge combination in the bipartite graph is a set of edges have no common nodes. If a particular weight sum of edges of one of all the edge combinations of one bipartite graph is largest, this particular edge combination is referred to as the maximum weight edge combination. The electronic device may traverse all edge combinations in the bipartite graph to find the maximum weight edge combination. The electronic device may also use a Kuhn-Munkres algorithm to solve the maximum weight edge combination of the bipartite graph. After the maximum weight edge combination is solved, the human head positions associated with the edges in the maximum weight edge combination can be used as the human head position detected in the image to be detected.
In this embodiment, since the intersecting human head positions may correspond to the same human head, the human head positions output by the convolutional neural network are mostly gathered near the actual human head position in the image to be detected. Therefore, the acquired human head positions (step 602, for example) and the selected human head positions (step 605, for example) are used as the nodes in the bipartite graph to construct the bipartite graph, and weights of the corresponding edges of the intersecting human head positions are reduced. By solving the maximum weight edge combination, the detected human head position in the image to be detected are acquired, and the human head detection can be performed more accurately.
In an embodiment, the image to be detected may be a video frame in a video, and the human head detection method further includes a step of performing human head tracking and performing people counting frame by frame. Referring to
S802: Perform human head tracking video according to the human head position detected in the image frame to be detected frame by video frame.
Specifically, after detecting the human head position in one video frame, the electronic device performs the human head tracking video frame by video frame by using the detected human head position as a starting point. The electronic device may specifically use a mean shift (average drift) tracking algorithm, an optical flow tracking algorithm, or a tracking-learning-detection (TLD) algorithm.
S804: Determine a moving direction and a positional relationship of the tracked human head position relative to a designated area.
The designated area refers to the area designated in the video frame. The moving direction of the tracked human head position relative to the designated area refers to that the human head position is, for example, moving toward or away from the designated area. The positional relationship of the tracked human head position relative to the designated area refers to that the human head position is inside or outside the designated area.
In an embodiment, when the tracked human head position crosses a line representing a boundary of the designated area in a direction toward the designated area, it is determined that the tracked human head position enters the designated area. When the tracked human head position crosses the line representing the boundary of the designated area in a direction away from the designated area, it is determined that the tracked human head position leaves the designated area.
In an embodiment, when the tracked human head position sequentially crosses a first line and a second line parallel with the first line, it is determined that the tracked human head position enters the designated area. When the tracked human head position sequentially crosses the second line and the first line, it is determined that the tracked human head position leaves the designated area.
The parallel first line and second line may be straight lines or curved lines. The designated area may be one of two areas formed by segmenting the image to be detected by the second line, without including the first line. In this embodiment, the moving direction and the positional relationship of the tracked human head position relative to the designated area are determined by the two lines, thereby preventing a judgment error caused by movement of the human head position in the vicinity of the boundary of the designated area, thereby ensuring the correctness of people counting.
S806: Perform people counting according to the determined moving direction and positional relationship.
The people counting may be specifically counting a combination of one or more of the number of accumulated people entering the designated area, the number of accumulated people leaving the designated area, and the dynamic number of people entering the designated area. Specifically, the electronic device may add 1 to the number of statistically accumulated people entering the designated area, and/or add 1 to the number of dynamic people entering the designated area when one tracked human head position enters the designated area. The electronic device may add 1 to the number of statistically accumulated people leaving the designated area, and/or subtract 1 from the number of dynamic people entering the designated area when one tracked human head position leaves the designated area
In this embodiment, the human head detection may applied to security application. The people counting is performed according to the moving direction and the positional relationship of the tracked human head position relative to the designated area. Based on accurate human head detection, the accuracy of people counting can be improved.
In an embodiment, the human head detection method further includes a step of detecting the human head position and continuing tracking near the human head position tracked in a previous video frame when the tracking of the human head position is interrupted. Referring to
S902: Track and record the human head position video frame by video frame.
Specifically, the electronic device tracks the detected human head position with the detected human head position in the image to be detected as a starting point, and records the tracked human head position.
S904: Acquire a human head position tracked in a previous recorded video frame if the tracking of the human head position in a current video frame is interrupted.
Specifically, when a character moves quickly or lighting changes, the tracking of the human head position may be interrupted, and in this case, the human head position tracked in the previous video frame and recorded during the tracking video frame by video frame is acquired.
S906: Detect human head positions in a local area covering the acquired human head position (in step 904) in the current video frame.
The local area covering the acquired human head position is smaller than a size of one video frame, and larger than a size of the area occupied by the human head position tracked in the previous video frame. A shape of the local area may be similar to a shape of the area occupied by the human head position tracked in the previous video frame. A center of the local area may overlap with a center of the area occupied by the human head position tracked in the previous video frame.
Specifically, the electronic device may detect the human head positions in the current video frame to find the human head positions belonging to the local area. The electronic device may also detect the human head positions only in the local area. The electronic device may specifically use the steps of steps S302 to S310 to detect the human head positions in the local area in the current video frame. The detected human head positions may be partially or entirely located in the local area. The electronic device may use the human head positions of which the centers are within the local area as the human head positions in the detected local area, and the human head positions of which the centers are outside the local area do not belong to the human head positions in the local area.
For example, when the human head position is represented by a position of a rectangular box including the human head image, if a width of the rectangular box tracked in the previous video frame is W and a height is H, a and b are set to coefficients greater than 1, then the local area may be the rectangular area having a width of a*W and a height of b*H and the same center as the rectangular box. If the center coordinates of the rectangular box tracked in the previous video frame are (X1, X2) and the center coordinates of another rectangular box indicating the human head position are (X2, Y2), then when |X1−X2|<W/2 and |Y1−Y2|<H/2, the rectangular box of which the center coordinates are (X2, Y2) is determined to be in the local area of the rectangular box of which the center coordinates are (X1, X2).
S908: Continue to perform step S902 from the human head position detected in the local area.
In this embodiment, when the tracking of the human head positions is interrupted, the human head positions can be detected from the vicinity of the human head positions detected in the previous frame, and the interrupted human head tracking can be recovered from the interruption and continued. The human head detection and the human head tracking are combined to ensure the continuity of the tracking. Further, the accuracy of people counting is ensured.
The specific principle of the foregoing human head detection method is described below with a specific application scenario. A large number of top view images at an elevator entrance scene are acquired in advance, and the human head positions in these top view images are marked or labeled. For example, a quadruple is used to indicate the position of the human head image in a rectangular box 1001 in
Referring to
Further, the electronic device performs human head tracking video frame by video frame according to the human head position detected in the image to be detected, and it is determined that a tracked human head position 1105 enters a designated area when the tracked human head position 1105 sequentially crosses a first line 1102 and a second line 1103 parallel with the first line 1102. When a tracked human head position 1106 sequentially crosses the second line 1103 and the first line 1102, it is determined that the tracked human head position 1106 leaves the designated area. The designated area in
In an embodiment, an electronic device is further provided, and an internal structure of the electronic device may be shown in
The segmenting module 1210 is configured to segment an image to be detected into one or more sub-images.
The convolutional neural network module 1220 is configured to segment the image to be detected into one or more sub-images; input each sub-image to a convolutional neural network trained according to training images having marked human head positions, and output, by a preprocessing layer including at least one convolutional layer and at least one pooling layer in the convolutional neural network, a first feature corresponding to each sub-image; map, through the convolutional layer after the preprocessing layer in the convolutional neural network, the first feature corresponding to each sub-image to a second feature corresponding to each sub-image; and map, through a regression layer of the convolutional neural network, the second feature corresponding to each sub-image to a human head position corresponding to each sub-image and a corresponding confidence level of the human head position.
The human head detection result determining module 1230 is configured to filter, according to the corresponding confidence level, the human head position corresponding to each sub-image, to acquire a human head position detected in the image to be detected.
According to the human head detection apparatus 1200, the convolutional neural network is trained in advance based on the training image having the marked human head position, and the convolutional neural network can automatically learn human head features. The trained convolutional neural network can automatically extract appropriate features from the sub-images to output candidate human head positions and corresponding confidence levels, and then filter, according to the confidence levels, to acquire the human head position in the image to be detected. The human head shape is not required to be assumed in advance, a missed detection caused by setting the human head shape can be avoided, and accuracy of the human head detection is improved. Moreover, in the convolutional neural network, the first features of the sub-images are output by the preprocessing layer including the convolutional layer and the pooling layer, and the second features are outputted by the convolutional layer after the preprocessing layer and before the regression layer to accurately describe human head features in the sub-images. Therefore, the second features are directly mapped to the human head positions and confidence levels by the regression layer, which is new application of the convolutional neural network of the new structure. Compared with the traditional circle detection, the accuracy of the human head detection is greatly improved.
In an embodiment, the segmenting module 1210 is further configured to segment the image to be detected into one or more sub-images of a fixed size, and adjacent sub-images in the segmented sub-images have an overlapping part. In this embodiment, there is an overlapping part between the adjacent segmented sub-images, thereby ensuring that the adjacent sub-images have stronger correlation, and improving accuracy of detecting a human head position from the image to be detected.
As shown in
The convolutional neural network adjusting module 1240 is configured to convert a fully connected layer after the preprocessing layer and before the regression layer included in the convolutional neural network for classification to a convolutional layer; and replace a regression layer in the convolutional neural network for classification with a regression layer configured to map the second feature output by the converted convolutional layer to the human head position and the corresponding confidence level.
The training module 1250 is configured to train the convolutional neural network including the preprocessing layer, the converted convolutional layer and the replaced regression layer by using the training image having the marked human head position.
In this embodiment, the training after reconstruction is performed based on the convolutional neural network for classification, to acquire the convolutional neural network for human head detection. The reconstruction of the convolutional neural network is not required, the training duration can be reduced and the efficiency of human head detection is improved.
In an embodiment, the convolutional neural network module 1220 is further configured to map, through a first convolutional layer in the regression layer in the convolutional neural network, the second feature corresponding to each sub-image to a human head position corresponding to each sub-image; and map, through a second convolutional layer in the regression layer in the convolutional neural network, the second feature corresponding to each sub-image to a confidence level corresponding to the output human head position.
As shown in
The filtering module 1231 is configured to screen, from the human head positions corresponding to the sub-images, to acquire a human head position corresponding to a confidence level greater than or equal to a confidence level threshold; and select a human head position intersecting with the acquired human head position in the image to be detected from the human head positions corresponding to the sub-images.
The human head position determining module 1232 is configured to determine, according to the acquired human head position and the selected human head position, the human head position detected in the image to be detected.
In this embodiment, the accuracy of the human head detection can be further improved by using the confidence levels and the intersection or not as the basis for determining the human head position in the image to be detected.
In an embodiment, the human head position determining module 1232 is further configured to use the acquired human head position and the selected human head position as nodes in a bipartite graph; assign default and positive weights to edges between the nodes in the bipartite graph; reduce the corresponding assigned weights when the human head positions indicated by the nodes associated with the edges intersect; and solve a maximum weight edge combination of the bipartite graph, to acquire the head position detected in the image to be detected.
In this embodiment, since the intersecting human head position are likely to correspond to the same human head, the human head positions output by the convolutional neural network are mostly gathered near the actual human head position in the image to be detected. Therefore, the acquired human head positions and the selected human head positions are used as nodes in the bipartite graph to construct the bipartite graph, and weights of the corresponding edges of the intersecting human head positions are relatively small. By solving the maximum weight edge combination, the human head position detected in the image to be detected are acquired, and the human head detection can be performed more accurately.
As shown in
a tracking module 1260, configured to perform head tracking video frame by video frame according to the human head position detected in the image to be detected;
a counting condition detecting module 1270, configured to determine a moving direction and a positional relationship of the tracked human head position relative to the designated area; and
a people counting module 1280, configured to perform people counting based on the determined moving direction and positional relationship.
In this embodiment, the human head detection is applied to the field of security. The people counting is performed according to the moving direction and the positional relationship of the tracked human head position relative to the designated area. Based on accurate human head detection, the accuracy of people counting can be ensured.
In an embodiment, the counting condition detecting module 1270 is further configured to: determine that the tracked human head position enters the designated area when the tracked human head position sequentially spans a first line and a second line parallel with the first line; and determine that the tracked human head position leaves the designated area when the tracked human head position sequentially span the second line and the first line.
In this embodiment, the moving direction and the positional relationship of the tracked human head position relative to the designated area are determined by two lines, thereby preventing a judgment error caused by the moving of the human head position near a boundary of the designated area, thereby ensuring the correctness of people counting.
As shown in
The tracking module 1260 is further configured to track and record the human head position video frame by video frame.
The human head position acquiring module 1290 is configured to acquire a human head position tracked in a previous recorded video frame if the tracking of the human head position in a current video frame is interrupted.
The convolutional neural network module 1220 is further configured to detect human head positions in a local area covering the acquired head position in the current video frame.
The tracking module 1260 is further configured to continue to perform the step of tracking and recording the human head position video frame by video frame from the human head positions detected in the local area.
In this embodiment, when the tracking of the human head positions is interrupted, the human head positions can be detected from the vicinity of the human head positions detected in the previous frame, and the interrupted human head tracking can be continued. The human head detection and the human head tracking are combined to ensure the continuity of the tracking. Further, the accuracy of people counting is ensured.
It should be understood that the steps in various embodiments of this application are not necessarily performed in an order indicated by the step numbers. Unless explicitly described in this specification, there is no strict sequence for execution of the steps. In addition, at least some steps in the embodiments may include a plurality of substeps or a plurality of stages. The substeps or the stages are not necessarily performed at a same moment, and instead may be performed at different moments. A performing sequence of the substeps or the stages is not necessarily performing in sequence, and instead may be performed in turn or alternately with another step or at least some of substeps or stages of the another step.
A person of ordinary skill in the art may understand that all or some of the processes of the methods in the foregoing embodiments may be implemented by a computer program instructing relevant hardware. The program may be stored in a non-volatile computer-readable storage medium. When the program runs, the processes of the foregoing methods in the embodiments are performed. The memory, storage, database or any other media in the embodiments provided in this application may include a non-volatile and/or volatile memory. The non-volatile memory may include: a read-only memory (ROM), a programmable ROM (PROM), an electrically programmable ROM (EPROM), an electrically erasable programmable ROM (EEPROM), or a flash memory. The volatile memory may include a random access memory (RAM) or an external cache memory. By way of illustration and not limitation, the RAM may be implemented in various forms, such as a static RAM (SRAM), a dynamic RAM (DRAM), a synchronous DRAM (SDRAM), a double data rate SDRAM (DDRSDRAM), an enhanced SDRAM (ESDRAM), a synchlink DRAM (SLDRAM), a rambus direct RAM (RDRAM), a direct rambus dynamic RAM (DRDRAM), and a rambus dynamic RAM (RDRAM).
Various technical features in the foregoing embodiments may be randomly combined. For ease of description, not all possible combinations of the various technical features in the foregoing embodiments are described. However, the combinations of the technical features should be considered as falling within the scope recorded in this specification as long as the combinations of the technical features are compatible with each other.
The foregoing embodiments only describe several implementations of this application, which are described specifically and in detail, and therefore cannot be construed as a limitation to the patent scope of the present disclosure. It should be noted that, a person of ordinary skill in the art may make various changes and improvements without departing from the ideas of this application, which shall all fall within the protection scope of this application. Therefore, the protection scope of the patent of this application shall be subject to the appended claims.
Claims
1. A method for detecting human head in an image performed by an electronic device comprising a processor, the method comprising:
- segmenting, by the electronic device, the image into one or more sub-images;
- inputting, by the electronic device, each sub-image to a convolutional neural network trained according to training images having marked human head positions, and outputting, by a preprocessing layer of the convolutional neural network comprising a first convolutional layer and a pooling layer, a first feature corresponding to each sub-image;
- mapping, by the electronic device through a second convolutional layer after the preprocessing layer in the convolutional neural network, the first feature corresponding to each sub-image to a second feature corresponding to each sub-image;
- mapping, by the electronic device through a regression layer in the convolutional neural network, the second feature corresponding to each sub-image to a human head position corresponding to each sub-image and a corresponding confidence level of the human head position; and
- filtering, by the electronic device according to the corresponding confidence level, human head positions corresponding to the one or more sub-images, to acquire detected human head positions in the image.
2. The method according to claim 1, wherein segmenting, by an electronic device, the image to be detected into one or more sub-images comprises:
- segmenting, by the electronic device, the image into one or more sub-images of a fixed size, wherein adjacent sub-images in the one or more sub-images partially overlap.
3. The method according to claim 1, wherein:
- a fully connected layer in a conventional convolution neural network is converted to the second convolutional layer;
- a conventional regression layer in a conventional convolutional neural network for image classification is replaced by the regression layer for mapping the second feature output by the second convolutional layer to the human head position and the corresponding confidence level; and
- the method further comprises training, by the electronic device, the convolutional neural network comprising the preprocessing layer, the second convolutional layer, and the regression layer by using the training images having the marked human head positions.
4. The method according to claim 1, wherein mapping, by the electronic device through the regression layer in the convolutional neural network, the second feature corresponding to each sub-image to a human head position corresponding to each sub-image and a corresponding confidence level of the human head position comprises:
- mapping, by the electronic device through a third convolutional layer in the regression layer of the convolutional neural network, the second feature corresponding to each sub-image to the human head position corresponding to each sub-image; and
- mapping, by the electronic device through a fourth convolutional layer in the regression layer of the convolutional neural network, the second feature corresponding to each sub-image to the confidence level corresponding to the human head position.
5. The method according to claim 1, wherein filtering, by the electronic device according to the corresponding confidence level, the human head positions corresponding to the one or more sub-images, to acquire the detected human head positions in the image comprises:
- screening, by the electronic device from the human head positions corresponding to the one or more sub-images, to acquire screened human head positions corresponding to confidence levels greater than or equal to a predetermined confidence level threshold;
- selecting, by the electronic device, human head positions intersecting with the screened human head positions from the screened human head positions to obtain overlapped human head positions; and
- determining, by the electronic device according to the screened human head positions and the overlapped human head positions, the detected human head positions of the image.
6. The method according to claim 5, wherein determining, by the electronic device according to the screened human head positions and the overlapped human head positions, the detected human head positions of the image comprises:
- using, by the electronic device, the screened human head positions and the overlapped human head positions as nodes in a bipartite graph;
- assigning, by the electronic device, default and positive weights to edges between the nodes in the bipartite graph;
- reducing, by the electronic device, weights of edges in the bipartite graph associated with the overlapped human head positions; and
- solving, by the electronic device, a maximum weight edge combination of the bipartite graph to obtain the detected human head positions of the image.
7. The method according to claim 1, wherein the image comprises a video frame in a video, and the method further comprises:
- performing, by the electronic device, human head tracking according to the detected human head positions video frame by video frame;
- determining, by the electronic device, a moving direction and a positional relationship of each of the tracked human head positions relative to a designated area; and
- performing, by the electronic device, people counting according to the moving direction and positional relationship of each of the tracked hu8man head positions.
8. The method according to claim 7, wherein determining, by the electronic device, the moving direction and the positional relationship of the tracked human head position relative to the designated area comprises:
- determining, by the electronic device, that the tracked human head position enters the designated area when the tracked human head position sequentially crosses a first line and a second line parallel with the first line; and
- determining, by the electronic device, that the tracked human head position leaves the designated area when the tracked human head position sequentially crosses the second line and the first line.
9. The method according to claim 7, wherein the method further comprises:
- tracking and recording, by the electronic device, the detected human head positions video frame by video frame;
- acquiring, by the electronic device, a human head position tracked in a previous video frame if the tracking of the human head position in a current video frame is interrupted;
- detecting, by the electronic device, a recovered human head position in the current video frame within a local area covering the acquired human head position in the previous video frame; and
- continuing, by the electronic device, tracking and recording the recovered human head position video frame by video frame.
10. An electronic device for detecting human head in an image, comprising a memory and a processor, the memory storing computer readable instructions, the computer readable instructions, when executed by the processor, causing the processor to perform the following steps:
- segmenting the image into one or more sub-images;
- inputting each sub-image to a convolutional neural network trained according to training images having marked human head positions, and outputting, by a preprocessing layer of the convolutional neural network comprising a first convolutional layer and a pooling layer, a first feature corresponding to each sub-image;
- mapping, through a second convolutional layer after the preprocessing layer in the convolutional neural network, the first feature corresponding to each sub-image to a second feature corresponding to each sub-image;
- mapping, through a regression layer in the convolutional neural network, the second feature corresponding to each sub-image to a human head position corresponding to each sub-image and a corresponding confidence level of the human head position; and
- filtering, according to the corresponding confidence level, human head positions corresponding to the one or more sub-images, to acquire detected human head positions in the image.
11. The electronic device according to claim 10, wherein segmenting, by an electronic device, the image into one or more sub-images comprises:
- segmenting the image into one or more sub-images of a fixed size, wherein adjacent sub-images in the one or more sub-images partially overlap.
12. The electronic device according to claim 10, wherein:
- a fully connected layer in a conventional convolution neural network is converted to the second convolutional layer;
- a conventional regression layer in a conventional convolutional neural network for image classification is replaced by the regression layer for mapping the second feature output by the second convolutional layer to the human head position and the corresponding confidence level; and
- the computer readable instructions causes the processor to further perform the step of training, by the electronic device, the convolutional neural network comprising the preprocessing layer, the second convolutional layer, and the regression layer by using the training images having the marked human head positions.
13. The electronic device according to claim 10, wherein mapping, through a regression layer in the convolutional neural network, the second feature corresponding to each sub-image to a human head position corresponding to each sub-image and a corresponding confidence level of the human head position comprises:
- mapping, through a third convolutional layer in the regression layer of the convolutional neural network, the second feature corresponding to each sub-image to the human head position corresponding to each sub-image; and
- mapping, through a fourth convolutional layer in the regression layer of the convolutional neural network, the second feature corresponding to each sub-image to the confidence level corresponding to the human head position.
14. The electronic device according to claim 10, wherein filtering, according to the corresponding confidence level, the human head position corresponding to the one or more sub-images, to acquire the detected human head positions in the image to be detected comprises:
- screening, from the human head positions corresponding to the one or more sub-images, to acquire screened human head positions corresponding to confidence levels greater than or equal to a predetermined confidence level threshold;
- selecting human head positions intersecting with the screened human head positions from the screened human head positions to obtain overlapped human head positions; and
- determining, according to the screened human head positions and the overlapped human head positions, the detected human head positions.
15. The electronic device according to claim 14, wherein determining, according to the screened human head positions and the overlapped human head positions, the detected human head positions in the image comprises:
- using the screened human head positions and the overlapped human head positions as nodes in a bipartite graph;
- assigning default and positive weights to edges between the nodes in the bipartite graph;
- reducing weights of edges in the bipartite graph associated with the overlapped human head positions; and
- solving a maximum weight edge combination of the bipartite graph to obtain the detected human head positions in the image.
16. The electronic device according to claim 10, wherein the image comprises a video frame in a video; and the computer readable instructions further causes the processor to perform the following steps:
- performing human head tracking according to the detected human head positions video frame by video frame;
- determining a moving direction and a positional relationship of each of the tracked human head positions relative to a designated area; and
- performing people counting according to the moving direction and positional relationship of each of the tracked hu8man head positions.
17. The electronic device according to claim 16, wherein determining the moving direction and the positional relationship of the tracked human head position relative to the designated area comprises:
- determining that the tracked human head position enters the designated area when the tracked human head position sequentially crosses a first line and a second line parallel with the first line; and
- determining that the tracked human head position leaves the designated area when the tracked human head position sequentially crosses the second line and the first line.
18. The electronic device according to claim 16, wherein the computer readable instructions further causes the processor to perform the following steps:
- tracking and recording the detected human head positions video frame by video frame;
- acquiring a human head position tracked in a previous video frame if the tracking of the human head position in a current video frame is interrupted;
- detecting a recovered human head position in the current video frame within a local area covering the acquired human head position in the previous video frame; and
- continuing tracking and recording the recovered human head position video frame by video frame.
19. A non-volatile storage medium for storing computer readable instructions, the computer readable instructions, when executed by one or more processors, causing the one or more processors to perform human head detection in an image by the following steps:
- segmenting the image into one or more sub-images;
- inputting each sub-image to a convolutional neural network trained according to training images having marked human head positions, and outputting, by a preprocessing layer of the convolutional neural network comprising a first convolutional layer and a pooling layer, a first feature corresponding to each sub-image;
- mapping, through a second convolutional layer after the preprocessing layer in the convolutional neural network, the first feature corresponding to each sub-image to a second feature corresponding to each sub-image;
- mapping, through a regression layer in the convolutional neural network, the second feature corresponding to each sub-image to a human head position corresponding to each sub-image and a corresponding confidence level of the human head position; and
- filtering, according to the corresponding confidence level, human head positions corresponding to the one or more sub-images, to acquire detected human head positions in the image.
20. The non-volatile storage medium according to claim 19, wherein segmenting the image into one or more sub-images comprises:
- segmenting the image into one or more sub-images of a fixed size, wherein adjacent sub-images in the one or more sub-images partially overlap.
Type: Application
Filed: Mar 12, 2019
Publication Date: Jul 4, 2019
Applicant: Tencent Technology (Shenzhen) Company Limited (Shenzhen)
Inventor: Deqiang JIANG (Shenzhen)
Application Number: 16/351,093