Plant intelligence-based PILLO underwater target detection algorithm

• Published in: Engineering applications of artificial intelligence 2023-11, Vol.126, p.106818, Article 106818
• Main Authors: Liu, Lizhao, Li, Pinrui
• Format: Article
• Language: English
• Subjects: Attention mechanism; PILLO; Plant intelligence; Underwater robot; Underwater target detection; YOLOv5
• ISSN: 0952-1976
• DOI: 10.1016/j.engappai.2023.106818

With the rapid development in the field of marine exploration, underwater robots play an increasingly important role in the military, fisheries, aquaculture, resource protection, etc. The main way for underwater robots to obtain information is underwater target detection, and the development of underwater robots is inseparable from the development of underwater target detection technology. Existing target detection algorithms have achieved good results on land, but are often unsatisfactory in actual underwater target detection. In this paper, based on the principle of plant intelligence and the target detection algorithm YOLOv5, the PILLO algorithm is improved to adapt to the underwater environment with blurred imaging and insufficient light and is specifically applied to underwater target detection. Specifically, the Swin Transformer is improved based on the principles of the multicore plant intelligence model, which then replaces part of the backbone network of the PILLO algorithm. The Shuffle Attention mechanism is improved based on the principles of the plant intelligence logic space attention transfer mechanism and added to the head network of the PILLO algorithm. The SIOU loss function was improved based on the principle of the ecological regulation module in the plant intelligence model and applied to the PILLO algorithm. The experimental results show that the PILLO model can improve the average accuracy (mAP) of underwater target detection by 87.1%, which is better than the common target detection model.