Lightweight Anchor-free Object Detection Algorithm Based on Keypoint Detection

GONG Hao-tian,ZHANG Meng

DOI: https://doi.org/10.11896/jsjkx.200700161

2021-01-01

Abstract:According to the large number of parameters of key-point object detection network and the problem of mismatching of bounding box,this paper proposes a lightweight key point anchor-free object detection algorithm.It inputs the image into the improved hourglass network to extract features,through the cascade corner pooling module and center pooling module,outputs three key points heatmap and their embedding vectors.At last,it matchs the key points by embedding vectors and draw the bounding box.The innovation of this paper is to applying the firemodule of SqueezeNet in the CenterNet object detection network,and replace the conventional convolution in the backbone with the depth separable convolution.At the same time,aiming at the mismatching bounding box problem in CenterNet,this algorithm adjusts the network's output and loss function.Experiment results show that the model size is reduced to 1/7 of CenterNet,while the accuracy and inference speed are still higher than the same size target detection algorithm like YOLOv3 and CornerNet-Lite.