Journal of System Simulation ›› 2020, Vol. 32 ›› Issue (7): 1279-1286.doi: 10.16182/j.issn1004731x.joss.19-VR0473

Previous Articles     Next Articles

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

Cai Xingquan, Tu Yuxin, Ge Yakun, Yang Zhe   

  1. School of Information Science and Technology, North China University of Technology, Beijing 100144, China
  • Received:2019-08-30 Revised:2019-11-18 Online:2020-07-25 Published:2020-07-15

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.

Key words: leaf recognition, feature map, CNN network, multitask loss function, region proposal

CLC Number: