Human head detection method, eletronic device and storage medium

Abstract

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.