Rapid and Accurate Crayfish Sorting by Size and Maturity Based on Improved YOLOv5
In response to the issues of high-intensity labor, low efficiency, and potential damage to crayfish associated with traditional manual sorting methods, an automated and non-contact sorting approach based on an improved YOLOv5 algorithm is proposed for the rapid sorting of crayfish maturity and size....
Main Authors: | , , , , , |
---|---|
Format: | Article |
Language: | English |
Published: |
MDPI AG
2023-07-01
|
Series: | Applied Sciences |
Subjects: | |
Online Access: | https://www.mdpi.com/2076-3417/13/15/8619 |
_version_ | 1797587064264327168 |
---|---|
author | Xuhui Ye Yuxiang Liu Daode Zhang Xinyu Hu Zhuang He Yan Chen |
author_facet | Xuhui Ye Yuxiang Liu Daode Zhang Xinyu Hu Zhuang He Yan Chen |
author_sort | Xuhui Ye |
collection | DOAJ |
description | In response to the issues of high-intensity labor, low efficiency, and potential damage to crayfish associated with traditional manual sorting methods, an automated and non-contact sorting approach based on an improved YOLOv5 algorithm is proposed for the rapid sorting of crayfish maturity and size. To address the difficulty in focusing on small crayfish, the Backbone is augmented with Coordinate Attention to boost its capability to extract features. Additionally, to address the difficulty in achieving high overall algorithm efficiency and reducing feature redundancy, the Bottleneck Transformer is integrated into both the Backbone and Neck, which improves the accuracy, generalization performance, and the model’s computational proficiency. The dataset of 3464 images of crayfish collected from a crayfish breeding farm is used for the experiments. The dataset is partitioned randomly, with 80% of the data used for training and the remaining 20% used for testing. The results indicate that the proposed algorithm achieves an mAP of 98.8%. Finally, the model is deployed using TensorRT, and the processing time for an image is reduced to just 2 ms, which greatly improves the processing speed of the model. In conclusion, this approach provides an accurate, efficient, fast, and automated solution for crayfish sorting. |
first_indexed | 2024-03-11T00:31:57Z |
format | Article |
id | doaj.art-100c6c0b3d414082ad0383b21bcaa8ff |
institution | Directory Open Access Journal |
issn | 2076-3417 |
language | English |
last_indexed | 2024-03-11T00:31:57Z |
publishDate | 2023-07-01 |
publisher | MDPI AG |
record_format | Article |
series | Applied Sciences |
spelling | doaj.art-100c6c0b3d414082ad0383b21bcaa8ff2023-11-18T22:35:17ZengMDPI AGApplied Sciences2076-34172023-07-011315861910.3390/app13158619Rapid and Accurate Crayfish Sorting by Size and Maturity Based on Improved YOLOv5Xuhui Ye0Yuxiang Liu1Daode Zhang2Xinyu Hu3Zhuang He4Yan Chen5School of Mechanical Engineering, Hubei University of Technology, Wuhan 430068, ChinaSchool of Mechanical Engineering, Hubei University of Technology, Wuhan 430068, ChinaSchool of Mechanical Engineering, Hubei University of Technology, Wuhan 430068, ChinaSchool of Mechanical Engineering, Hubei University of Technology, Wuhan 430068, ChinaSchool of Mechanical Engineering, Hubei University of Technology, Wuhan 430068, ChinaSchool of Mechanical and Electric Engineering, Wuhan Donghu University, Wuhan 430212, ChinaIn response to the issues of high-intensity labor, low efficiency, and potential damage to crayfish associated with traditional manual sorting methods, an automated and non-contact sorting approach based on an improved YOLOv5 algorithm is proposed for the rapid sorting of crayfish maturity and size. To address the difficulty in focusing on small crayfish, the Backbone is augmented with Coordinate Attention to boost its capability to extract features. Additionally, to address the difficulty in achieving high overall algorithm efficiency and reducing feature redundancy, the Bottleneck Transformer is integrated into both the Backbone and Neck, which improves the accuracy, generalization performance, and the model’s computational proficiency. The dataset of 3464 images of crayfish collected from a crayfish breeding farm is used for the experiments. The dataset is partitioned randomly, with 80% of the data used for training and the remaining 20% used for testing. The results indicate that the proposed algorithm achieves an mAP of 98.8%. Finally, the model is deployed using TensorRT, and the processing time for an image is reduced to just 2 ms, which greatly improves the processing speed of the model. In conclusion, this approach provides an accurate, efficient, fast, and automated solution for crayfish sorting.https://www.mdpi.com/2076-3417/13/15/8619crayfish sortingYolov5attention mechanismmaturitysizeTensorRT |
spellingShingle | Xuhui Ye Yuxiang Liu Daode Zhang Xinyu Hu Zhuang He Yan Chen Rapid and Accurate Crayfish Sorting by Size and Maturity Based on Improved YOLOv5 Applied Sciences crayfish sorting Yolov5 attention mechanism maturity size TensorRT |
title | Rapid and Accurate Crayfish Sorting by Size and Maturity Based on Improved YOLOv5 |
title_full | Rapid and Accurate Crayfish Sorting by Size and Maturity Based on Improved YOLOv5 |
title_fullStr | Rapid and Accurate Crayfish Sorting by Size and Maturity Based on Improved YOLOv5 |
title_full_unstemmed | Rapid and Accurate Crayfish Sorting by Size and Maturity Based on Improved YOLOv5 |
title_short | Rapid and Accurate Crayfish Sorting by Size and Maturity Based on Improved YOLOv5 |
title_sort | rapid and accurate crayfish sorting by size and maturity based on improved yolov5 |
topic | crayfish sorting Yolov5 attention mechanism maturity size TensorRT |
url | https://www.mdpi.com/2076-3417/13/15/8619 |
work_keys_str_mv | AT xuhuiye rapidandaccuratecrayfishsortingbysizeandmaturitybasedonimprovedyolov5 AT yuxiangliu rapidandaccuratecrayfishsortingbysizeandmaturitybasedonimprovedyolov5 AT daodezhang rapidandaccuratecrayfishsortingbysizeandmaturitybasedonimprovedyolov5 AT xinyuhu rapidandaccuratecrayfishsortingbysizeandmaturitybasedonimprovedyolov5 AT zhuanghe rapidandaccuratecrayfishsortingbysizeandmaturitybasedonimprovedyolov5 AT yanchen rapidandaccuratecrayfishsortingbysizeandmaturitybasedonimprovedyolov5 |