YOLOX-Mobile: A Target Detection Algorithm More Suitable for Mobile Devices
| Autor: | Songzhe Ma, Huimin Lu, Yifan Wang, Han Xue |
|---|---|
| Rok vydání: | 2022 |
| Předmět: | |
| Zdroj: | Journal of Physics: Conference Series. 2203:012030 |
| ISSN: | 1742-6596 1742-6588 |
| DOI: | 10.1088/1742-6596/2203/1/012030 |
| Popis: | With the continuous development of network, the increasing popularity of embedded technology, and the continuous improvement of mobile terminal computing ability, how to reduce the size of the model without affecting the accuracy as much as possible, and how to deploy the algorithm in embedded devices has become the current research hotspot. In this paper, we propose a new target detection algorithm called YOLOX-Mobile. Firstly, in order to further reduce the power consumption of embedded devices while maintaining the performance, we replaced the SiLU activation function in the original YOLOX with Mish activation function. Because SiLU activation functions occupy a certain amount of computing and storage resources, and the cost of calculating the Sigmoid type functions on mobile devices is much higher. Therefore, we use Mish functions that are smoother, non-monotonic, unbounded upper and lower bounds are used as activation functions to better meet the low power requirements of embedded devices. Secondly, we used Focal Loss as the Loss function of obj_output to achieve the balance of positive and negative samples, as well as hard-to-classify and easy-to-classify samples. Thirdly, we introduce the Involution operator and use it as the convolution kernel of 3×3. We validated the proposed algorithm on public dataset VOC2012 against the current mainstream YOLOX, YOLOX-M, and YOLOX-S algorithms. The mAP of our proposed algorithm is 78.22% and the detection speed is 55.26 FPS. Compared with the original YOlOX, YOLOX-M and YOLOX-S algorithms, the average FPS is improved by 1.99% and 4.13 FPS, and the average Params Size is reduced by 28.80%. Experimental results show that our proposed algorithm improves the accuracy and speed of detection on top of greatly reducing the network parameters and computation, making it a more suitable target detection algorithm for mobile devices. |
| Databáze: | OpenAIRE |
| Externí odkaz: |
|