Target Instance Segmentation Algorithm in Accurate Location for UAV
ZHU Hui-Jie1 WANG Yong1 ZHAO Zhen-Yu1 ZHANG Yao-Qin2 LIU Li-Hang1
1. Science and Technology on Near-surface Detection Laboratory, Wuxi Jiangsu 214035, China 2. North Hu-guang Optics Electronics Co.Ltd, Wuxi Jiangsu, 214035, China
Abstract：Unmanned aerial vehicle (UAV) mainly relies on monocular camera in target detection and location. The existing technology exists inaccurate target segmentation and low accuracy in location. In order to improve the performance of application, this paper introduces the instance segmentation algorithm SOLO V2, which has excellent performance and eciency, to segment the target accurately. In order to solve the problems of lack of training samples, target rotation and large scale variation, some methods including data enhancement, multi-scale training and network structure optimization are used. By training and testing the actual pictures, the results show that the algorithm can detect the target more accurately, and basically meet the actual needs. Compared with the original SOLO V2, the accuracy and recall rate are both greatly improved. Finally, some pictures are downloaded from the Internet for testing, and many targets can be detected robustly, which further verifies the e ectiveness of the algorithm.
2021, Vol. 7 
