A Tracking Method Based on Lightweight Feature Extraction Network for Tracking Robot

Han Hu,Gang Yu,Geyang Wu,Tongda Sun
DOI: https://doi.org/10.1007/978-3-030-89134-3_34
2021-01-01
Abstract:This paper proposes to use lightweight feature extraction network as backbone of tracking network applied in embeded system of tracking robot, which achieves speed improvement and similar accuracy as compared with other mainstream tracking algorithms. Firstly, since there are unbalance between hard and easy samples on training dataset, data augmentation is proposed to increase hard samples in training phase. Secondly, modified Ghostnet is used as feature extraction network of tracking network, which dramatically decreases the float point operation and model parameters and achives realtime speed to meet the demand of embeded systems. Thirdly, focal loss is used as classification loss function instead of BCE loss, because BCE fuction cannot take unbalance between position and negative samples into consideration, and DIOU loss is taken as regression loss function to train tracking network, which can regress predicted box close to ground truth box. Experimental resutls show that the proposed method has better performance than traditional methods.
What problem does this paper attempt to address?