Optimierung der Forstschädlingsdetektion basierend auf Deep Learning

ZHAO Yan ,  

LIU Ying-an ,  

YE Qiao-lin ,  

ZHOU Xiao-liang ,  

摘要

Derzeit basieren die meisten Forschungen zur Erkennung von Forstschädlingen auf traditionellen maschinellen Lernalgorithmen, die Probleme wie geringe Genauigkeit und schlechte Leistung aufweisen. Daher wurde ein auf dem Deep-Learning-Modell YOLOv4 basierendes Forstschädlings-Erkennungsmodell namens Pest-YOLOv4 vorgeschlagen. Der K-means++-Algorithmus wurde verwendet, um Prioritätsrahmen zu clustern und den durchschnittlichen IoU-Wert der Prioritätsrahmen zu erhöhen. Die Methoden ECA (Efficient Channel Attention) und CBAM (Convolutional Block Attention Module) wurden kombiniert, um den ECA-CBAM-Aufmerksamkeitsmechanismus zu bilden, der das Netzwerk dazu bringt, sich stärker auf für die Schädlingsdetektion relevante Merkmale zu konzentrieren. Der Netzwerkrumpf wurde neu organisiert und bildet SPP-PANet, das Merkmalsinformationen aus mehreren Rezeptorfeldern integriert. Die Verlustfunktion wurde mithilfe des Focal-Loss-Ansatzes verbessert, um das Verhältnis zwischen positiven und negativen Beispielen auszubalancieren und dabei schwer unterscheidbare Beispiele beim Lernen zu berücksichtigen. Die Experimentergebnisse zeigen, dass das Pest-YOLOv4-Forstschädlings-Erkennungsmodell eine mAP von 90,4 % erreicht, was eine Steigerung um 4,2 % gegenüber YOLOv4 bedeutet. Die FPS bleibt bei 33,4 f/s und erfüllt die Anforderungen an Genauigkeit und Echtzeitfähigkeit bei der Forstschädlingsdetektion.

关键词

Forstschädlingsdetektion;Deep Learning;Pest-YOLOv4;Aufmerksamkeitsmechanismus;Focal Loss

阅读全文

以上内容由讯飞翻译自动生成,翻译内容仅供参考。对于因使用本网站翻译内容产生的相关后果,本网站不承担任何商业和法律责任。

The above content is generated by Large Model Translation. The translated content is for reference only. We do not assume any commercial or legal responsibilty for any consequences arising from the use of our website