arXiv reaDer
オープンフィールド環境でのロボット収穫のための改良された YOLOv5s アーキテクチャに基づくリアルタイムのイチゴ検出
Real-time Strawberry Detection Based on Improved YOLOv5s Architecture for Robotic Harvesting in open-field environment
この研究では、屋外環境でイチゴを検出するための YOLOv5 ベースのカスタム オブジェクト検出モデルを提案しました。 YOLOv5s の元のアーキテクチャは、バックボーン ネットワーク内の C3 モジュールを C2f モジュールに置き換えることによって変更され、より優れた機能勾配フローが提供されました。次に、この研究では、YOLOv5s のバックボーン ネットワークの最終層にある Spatial Pyramid Pooling Fast を Cross Stage Partial Net と組み合わせて、イチゴ データセットに対する汎化能力を向上させました。提案されたアーキテクチャは YOLOv5s-Straw と名付けられました。 3 つの成熟度クラス (未熟、ほぼ成熟、成熟) を持つイチゴの樹冠の RGB 画像データセットは、オープンフィールド環境で収集され、輝度の低下、輝度の増加、ノイズの追加を含む一連の操作を通じて強化されました。オープンフィールド環境におけるイチゴ検出のための提案された方法の優位性を検証するために、4 つの競合検出モデル (YOLOv3-tiny、YOLOv5s、YOLOv5s-C2f、および YOLOv8s) がトレーニングされ、同じ計算環境下でテストされ、YOLOv5s-と比較されました。ストロー。結果は、提案されたアーキテクチャを使用すると最高の平均精度である 80.3% が達成されたのに対し、YOLOv3-tiny、YOLOv5s、YOLOv5s-C2f、および YOLOv8s では同様の精度がそれぞれ 73.4%、77.8%、79.8%、79.3% であったことが示されました。 。具体的には、YOLOv5s-Straw の平均精度は、未成熟クラスで 82.1%、ほぼ成熟クラスで 73.5%、成熟クラスで 86.6% であり、最新の YOLOv8s よりもそれぞれ 2.3% と 3.7% 高くなりました。 。このモデルには、画像あたり 18 ミリ秒の推論速度を持つ 8.6*10^6 のネットワーク パラメーターが含まれていましたが、YOLOv8s の推論速度は 21.0 ミリ秒と遅く、重いパラメーターは 11.1*10^6 でした。これは、提案されたモデルが高速であることを示しています。ロボットピッキングのためのリアルタイムのイチゴ検出と位置特定には十分です。
This study proposed a YOLOv5-based custom object detection model to detect strawberries in an outdoor environment. The original architecture of the YOLOv5s was modified by replacing the C3 module with the C2f module in the backbone network, which provided a better feature gradient flow. Secondly, the Spatial Pyramid Pooling Fast in the final layer of the backbone network of YOLOv5s was combined with Cross Stage Partial Net to improve the generalization ability over the strawberry dataset in this study. The proposed architecture was named YOLOv5s-Straw. The RGB images dataset of the strawberry canopy with three maturity classes (immature, nearly mature, and mature) was collected in open-field environment and augmented through a series of operations including brightness reduction, brightness increase, and noise adding. To verify the superiority of the proposed method for strawberry detection in open-field environment, four competitive detection models (YOLOv3-tiny, YOLOv5s, YOLOv5s-C2f, and YOLOv8s) were trained, and tested under the same computational environment and compared with YOLOv5s-Straw. The results showed that the highest mean average precision of 80.3% was achieved using the proposed architecture whereas the same was achieved with YOLOv3-tiny, YOLOv5s, YOLOv5s-C2f, and YOLOv8s were 73.4%, 77.8%, 79.8%, 79.3%, respectively. Specifically, the average precision of YOLOv5s-Straw was 82.1% in the immature class, 73.5% in the nearly mature class, and 86.6% in the mature class, which were 2.3% and 3.7%, respectively, higher than that of the latest YOLOv8s. The model included 8.6*10^6 network parameters with an inference speed of 18ms per image while the inference speed of YOLOv8s had a slower inference speed of 21.0ms and heavy parameters of 11.1*10^6, which indicates that the proposed model is fast enough for real time strawberry detection and localization for the robotic picking.
updated: Tue Aug 29 2023 01:57:19 GMT+0000 (UTC)
published: Tue Aug 08 2023 02:28:48 GMT+0000 (UTC)
参考文献 (このサイトで利用可能なもの) / References (only if available on this site)
被参照文献 (このサイトで利用可能なものを新しい順に) / Citations (only if available on this site, in order of most recent)
Amazon.co.jpアソシエイト