用合成数据训练目标检测模型,自动统计无人机拍摄的椰子树数量。
Coconut Palm Tree Counting on Drone Images with Deep Object Detection and Synthetic Training Data
- 基于YOLOv7构建检测模型,结合合成图像增强训练数据。
- 通过调整参数和飞行高度,mAP从0.65提升至0.88。
- 适合农业管理者用于高效估算椰子树数量,减少人工成本。
无人机已广泛应用于农业等领域。本文利用YOLO实时目标检测模型,在加纳农场的无人机影像中识别并统计椰子树数量。由于种植阶段多样,该农场已无法准确掌握树苗数量。人工清点耗时且易出错,而精确计数对优化产量和预测生产至关重要。研究采用半自动化框架评估了YOLO在棕榈树检测中的表现,探索了精度提升策略,并探讨其对农民的实际价值。数据采集于2022年9月。为应对真实数据稀缺问题,通过将真实影像中的树木重新定位到合成背景图像中生成训练数据。使用在COCO数据集(不含椰子树)上预训练的YOLOv7模型,经定制化数据微调后进行测试。实验中优化了超参数,显著提升了平均精度均值(mAP)。同时测试了不同飞行高度,以确定最优拍摄高度。初始[email protected]为0.65,优化后达0.88,验证了合成数据在农业场景中的有效性。
原文摘要 · Abstract (English)
Drones have revolutionized various domains, including agriculture. Recent advances in deep learning have propelled among other things object detection in computer vision. This study utilized YOLO, a real-time object detector, to identify and count coconut palm trees in Ghanaian farm drone footage. The farm presented has lost track of its trees due to different planting phases. While manual counting would be very tedious and error-prone, accurately determining the number of trees is crucial for efficient planning and management of agricultural processes, especially for optimizing yields and predicting production. We assessed YOLO for palm detection within a semi-automated framework, evaluated accuracy augmentations, and pondered its potential for farmers. Data was captured in September 2022 via drones. To optimize YOLO with scarce data, synthetic images were created for model training and validation. The YOLOv7 model, pretrained on the COCO dataset (excluding coconut palms), was adapted using tailored data. Trees from footage were repositioned on synthetic images, with testing on distinct authentic images. In our experiments, we adjusted hyperparameters, improving YOLO's mean average precision (mAP). We also tested various altitudes to determine the best drone height. From an initial [email protected] of $0.65$, we achieved 0.88, highlighting the value of synthetic images in agricultural scenarios.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。