Potato Bud Detection with Improved Faster R-CNN

摘要

This article proposes an improved Faster R-CNN model to achieve better detection performance for potato buds, with the goal of preparing for the automated cutting of seed potatoes. Detection results ofFaster R-CNNs with eight pretrained networks werecompared, and ResNet50 was adopted as the backbone network in Faster R-CNN. On this basis, three model strategies, including feature concatenation, tweaks ofResNet50, and modification of the default anchors with the chaos optimization-based k-means algorithm, were proposed to improve the detection performance for potato buds. Experimental results on the test set demonstrated that the improved Faster R-CNN achieved an average precision (AP) of 97.71 %, which is 5.98% higher than that of the original Faster R-CNN and 14.38% higher than that of YOLOv2. In addition, the average running time per image with the improved Faster R-CNN was 0.166 s, the same as that of the original Faster R-CNN. In other words, the improved Faster R-CNN greatly boosted the detection performance for potato buds without incurring any noticeable additional computational overhead, thus satisfying the requirements for real-time processing. Consequently, the improved Faster R-CNN can provide a solidfoundation for the automated cutting of seed potatoes.