Graph Pooling based Human Detection Method for Industrial Application

Y Shao, Y Zhao, H Yu, M Liu… - 2023 26th International …, 2023 - ieeexplore.ieee.org
Y Shao, Y Zhao, H Yu, M Liu, Y Ma
2023 26th International Conference on Computer Supported …, 2023ieeexplore.ieee.org
Human detection is an important problem which has many applications in collaborative
manufacturing, such as employee counting and human fatigue detection. Recently, some
methods based on convolutional neural network (CNN) have been proposed to solve this
problem, but they often suffer from poor generalization due to the lack of real-world datasets
from industrial scenarios. In this paper, we propose a graph pooling based human detection
method to improve the generalization of existing CNN models. First, we segment an image …
Human detection is an important problem which has many applications in collaborative manufacturing, such as employee counting and human fatigue detection. Recently, some methods based on convolutional neural network (CNN) have been proposed to solve this problem, but they often suffer from poor generalization due to the lack of real-world datasets from industrial scenarios. In this paper, we propose a graph pooling based human detection method to improve the generalization of existing CNN models. First, we segment an image and abstract it as a graph where nodes represent the segmented regions and the features of nodes are pixel values of the segmentation. Second, a graph pooling model is used to extract coarse-grained but informative features from the graph. Next, the pooled graph is reconstructed to image and input to a popular CNN model (i.e., YOLOv5) for human detection. We perform experiments on data containing a great number of images from industrial scenarios. Experimental results show that the proposed method outperforms the popular CNN model and has better generalization performance.
ieeexplore.ieee.org
Showing the best result for this search. See all results