مجله ماشین بینایی و پردازش تصویر

مجله ماشین بینایی و پردازش تصویر

یک معماری جدید از شبکه‌YOLOv5 با بکارگیری مکانیسم توجه برای بهبود مصالحه دقت-سرعت در آشکارسازی میوه سیب

نوع مقاله : مقاله پژوهشی

نویسندگان
1 دانش‌آموخته کارشناسی ارشد مهندسی برق گرایش مخابرات-سیستم، دانشگاه بین‌المللی امام خمینی(ره)، قزوین، ایران
2 دانشکده فنی و مهندسی، گروه مهندسی برق، دانشگاه بین‌المللی امام خمینی(ره)، قزوین، ایران
چکیده
آشکارسازی میوه با توجه به شرایط روشنایی متفاوت، انسداد و همپوشانی یک کار چالش‌برانگیز در ربات‌های برداشت مبتنی بر بینایی ماشین است. هدف از این مقاله بهبود مصالحه دقت-سرعت در آشکارسازی میوه سیب در سیستم بینایی ربات‌های برداشت کننده کشاورزی است. با توجه به کاربردهای اخیر ماژول‌های توجه در زمینه آشکارسازی شیء، معماری جدیدی از شبکه YOLOv5 پیشنهاد شده است که در آن ماژول توجه کانالی ECA در ستون فقرات شبکه، جایگزین ماژول C3 شده است. ماژول ECA علی‌رغم کاهش تعداد پارامترهای شبکه اثر قابل توجهی در کارایی آشکارسازی نداشت و با افزایش سرعت به میزان% 22نسبت به YOLOv5 نسخه نانو، توانست مصالحه بهتری بین دقت و سرعت برقرار کند. برای ارزیابی معماری پیشنهادی از سه نوع مجموعه داده KFuji، MinneApple و ACFR در مرحله آموزش و آزمون استفاده شد و در حالتی که پایگاه داده آموزش و آزمون یکی نبودند، روش‌ یادگیری انتقالی برای بهبود نتایج آزمون به کار گرفته شد. در حالتی که داده‌های آموزش و آزمون یکی بودند، استفاده از معماری پیشنهادی منجر به بهبود نسبی عدد مصالحه به میزان 21.2% در مقایسه با ماژول C3 شد و در حالت یادگیری انتقالی که داده‌های آموزش و آزمون یکی نبودند، بهبود نسبی 18% در عدد مصالحه به دست آمد.
کلیدواژه‌ها

عنوان مقاله English

A novel architecture of YOLOv5 network using attention mechanism to improve accuracy-speed trade-off of apple fruit detection

نویسندگان English

marziye mahmoudifar 1
Neda Faraji 2
1 Graduated of Electrical Engineering, Dept. of Engineering, Imam Khomeini International University
2 Electrical Engineering Department, Faculty of Engineering and Technology, Imam Khomeini International University, Qazvin, Iran
چکیده English

Fruit detection due to different lighting conditions, occlusion and overlap is a challenging task in machine vision-based harvesting robots. The aim of this article is to improve the accuracy-speed trade-off in apple fruit detection in the vision system of agricultural harvesting robots. Considering the recent applications of attention modules in the field of object detection, we have proposed a new architecture of YOLOv5 network in which the ECA channel attention module is replaced by the C3 module in the backbone of the network. Despite reducing the number of network parameters, the ECA module has not had a significant effect on the detection efficiency, and by increasing the speed by 22% compared to the YOLOv5 Nano version, it has been able to establish a better trade-off between accuracy and speed. To evaluate the proposed architecture, three datasets KFuji, MinneApple and ACFR are used in the training and testing phase, and in the case that the training and testing databases are not the same, the transfer learning method is used to improve the test results. In the case where the training and test data are the same, the use of the proposed architecture leads to a relative improvement of the trade-off by 21.2% compared to the C3 module. In the case of transfer learning where the training and test data are not the same, a relative improvement of 18% in the trade-off has been achieved.

کلیدواژه‌ها English

YOLOv5
detection
accuracy-speed trade-off
attention module
transfer learning
دوره 10، شماره 4
زمستان 1402
صفحه 1-14

  • تاریخ دریافت 07 مرداد 1402