Open AccessDesign of lightweight pedestrian detection network in railway scenes
Author(s) -
Chuanyi Han,
Jinlei Zhu,
Feng Li,
Shengkai Wan,
Houjin Chen
Publication year - 2020
Publication title -
journal of physics. conference series
Language(s) - English
Resource type - Journals
SCImago Journal Rank - 0.21
H-Index - 85
eISSN - 1742-6596
pISSN - 1742-6588
DOI - 10.1088/1742-6596/1544/1/012053
Subject(s) - pruning , pedestrian detection , pedestrian , convolutional neural network , computer science , reduction (mathematics) , pyramid (geometry) , feature (linguistics) , artificial intelligence , object detection , artificial neural network , computer vision , pattern recognition (psychology) , engineering , mathematics , transport engineering , linguistics , philosophy , geometry , agronomy , biology
The deployment of many advanced pedestrian detection applications is largely hindered by the high computational cost of deep convolutional neural networks (CNNs). In this paper, we propose a two-step pruning method to design a lightweight pedestrian detection network in railway scenes. The first step is feature pyramid network (FPN) pruning, which utilizes the characteristic of pedestrian in railway scenes and the FPN structure in YOLOv3. The second step is regular channel pruning, which utilizes network slimming knowledge and is an accelerator-friendly pruning strategy. Our two-step pruning method gives about 88% reduction in parameters and about 74% reduction in computing complexity with comparable detection accuracy.