基于改进YOLOv3-Tiny的番茄苗分级检测

为了提高番茄苗分选移栽分级检测精度,该研究提出了YOLOv3-Tiny目标检测改进模型.首先建立了番茄穴盘苗数据集,使用K-means++算法重新生成数据集锚定框,提高网络收敛速度和特征提取能力;其次为目标检测模型添加SPP空间金字塔池化,将穴孔局部和整体特征融合,提高了对弱苗的召回率;同时加入路径聚合网络(PANet...

Tomato seedling classification detection using improved YOLOv3-Tiny

Here, an improved YOLOV3-TINY target detection model was proposed to enhance the detection accuracy of the seedling classification and detection in the process of tomato seedling transplanting. First of all, 2160 images of the individual tomato seedling were collected to pre-process the image of the tomato hole. A LabelImg software was then selected to mark the image. After that, the data enhancement was performed on the image, such as the rotate and flip operations. As such, 25080 images were generated, where 22,800 images were taken as the training set, and 2280 images were the test set. A target detection model of tomato hole seedling was improved for the better convergence speed of the network and the feature extraction, where the K-means ++ was used to regenerate the anchors of the tomato plug seedling dataset. Secondly, a spatial pyramid pooling (SPP) was added into the target detection model, further to integrate the local and global features of the plug holes for the less recall rate of weak seedlings. A path aggregation network (PANet) was also added to improve the fine-grained detection. A spatial attention mechanism (SAM) was then introduced to reduce the background noise in the target detection model. An adaptive feature fusion network was selected to directly learn the features from the other levels, where spatial filtering was performed for better features fusion. A CIoU loss function strategy was adopted to improve the convergence of the model. Eventually, the model training was conducted in a computer-deep learning environment after the dataset production and network construction. The results show that the Mean Average Precision value reached 97.64%, which was higher than 94.17% of the original. The F1 value of the improved YOLOv3-Tiny reached 0.94, which was higher than 0.92 of the original. A comparative experiment was also performed on the different types of tomato plug seedlings, further to verify the effectiveness and feasibility of the improved model. It was found that the improved YOLOv3-Tiny target detection model was fully met the requirements of tomato plug seedling grading detection, where Average Precision values were 98.22%, 94.69%, and 99.99% for the strong, weak, and no seedlings, respectively. Additionally, the improved network structure and the training strategy were used to verify the model in the process of the ablation experiment. We found that every improvement method of the model in the research has positive significance, and the introduction of Panet has the most obvious improvement in the Mean Average Precision value of the model, which is increased by 2.17 percentage points. Using the improved YOLOv3-Tiny target detection algorithm to compare with target detection algorithms such as YOLOv3-Tiny, Faster-RCNN and CenterNet, it is found that under the condition that the overlap threshold is 50%, the Mean Average Precision of the improved YOLOv3-Tiny The value is still 0.47 percentage points higher than other CenterNet algorithms with the highest Mean Average Precision value; the improved YOLOv3-Tin detection speed is 5.03 ms per image, which is 8.39ms higher than the YOLOv3 large target detection algorithm with the highest detection speed.The finding can also provide a strong reference to detect the seedling sorting during tomato production.