Real-time Leaf Recognition Method Based on CNN Network and Multi-task Loss Function

Authors

Xingquan Cai, School of Information Science and Technology, North China University of Technology, Beijing 100144, China;
Yuxin Tu, School of Information Science and Technology, North China University of Technology, Beijing 100144, China;
Yakun Ge, School of Information Science and Technology, North China University of Technology, Beijing 100144, China;
Zhe Yang, School of Information Science and Technology, North China University of Technology, Beijing 100144, China;

Abstract

Abstract: Aiming at the problems that the traditional leaf recognition is susceptible to the environmental interference and is difficult to realize the multi-leaf real-time recognition in complex background, a real-time leaf recognition method based on CNN network and multi-task loss function is proposed. The CNN network is used to the extract image feature maps and input them into RPN network to generate regional proposals. On the basis of the feature maps and region proposals, the feature map is proposaled, the leaf classification and bounding box regression are performed respectively, and the leaf classification and position of the leaf prediction box are predicted. The multi-task loss function is used to constrain the classification and regression to improve the accuracy and speed of the leaf image classification and regression. Experimental results show that the average real-time leaf recognition accuracy is 91.8%, and the average real-time leaf recognition speed is 25 fps.

Recommended Citation

Cai, Xingquan; Tu, Yuxin; Ge, Yakun; and Yang, Zhe (2020) "Real-time Leaf Recognition Method Based on CNN Network and Multi-task Loss Function," Journal of System Simulation: Vol. 32: Iss. 7, Article 8.
DOI: 10.16182/j.issn1004731x.joss.19-VR0473
Available at: https://dc-china-simulation.researchcommons.org/journal/vol32/iss7/8

First Page

1279

Revised Date

2019-11-18

DOI Link

https://doi.org/10.16182/j.issn1004731x.joss.19-VR0473

Last Page

1286

CLC

TP391.9

Recommended Citation

Cai Xingquan, Tu Yuxin, Ge Yakun, Yang Zhe. Real-time Leaf Recognition Method Based on CNN Network and Multi-task Loss Function[J]. Journal of System Simulation, 2020, 32(7): 1279-1286.

DOI

10.16182/j.issn1004731x.joss.19-VR0473