Improved tongue detection algorithm based on YOLO V4

Abstract

To address the problem of target recognition of the tongue body during tongue image extraction. A method to improve YOLO V4 target detection is proposed to detect the tongue. The tongue region of the image is detected by the YOLO V4 target detection algorithm combining MobileNet V1 with an attention mechanism. The network combining MobileNet V1 can well reduce the size of the model, but may reduce the accuracy, which can be well compensated by imposing an attention mechanism. The experimental results show that the proposed algorithm achieves 91.98% mAP on tongue detection, which is 3.66% higher compared to YOLO V4, 17.7% higher single detection speed, and 80.7% reduction in the amount of parameters of the model, which has a better detection and localization effect, and has some guiding significance for the subsequent research on embedded devices for tongue diagnosis.