Strawberry Ripeness Assessment Via Camouflage-Based Data Augmentation for Automated Strawberry Picking Robot
Abstract
Vision-based strawberry picking and placing is one of the main objectives for strawberry harvesting robots to complete visual servoing procedures accurately. Occlusion is the main challenge in strawberry ripeness detection for agriculture robots. In this study, strawberry ripeness detection was proposed using a camouflage-based data augmentation strategy to simulate the natural environment of strawberry harvesting conditions. Yolov4, Yolov4 tiny and Yolov4 scaled, and their traditional data augmentation and camouflage-based data augmentation derivatives were used to find out the effect of camouflage-based augmentation technique in overcoming the occlusion issue. Then the results were mainly evaluated based on mean Intersection over Union (IoU), F-1 score, average precision (AP) for ripe and unripe strawberries and frame per second (fps). Yolov4 tiny with camouflage-based data augmentation technique has demonstrated superior performance in detecting ripe and unripe strawberries with 84% IoU accuracy ~99% AP for ripe and unripe strawberries at an average of 206-fps, satisfying the agriculture strawberry harvesting robot operation need. The performance of the suggested technique was then tested successfully using a dataset termed the challenge dataset in this study to demonstrate its performance in a complex and occluded strawberry harvesting environment. Camouflage-based data augmentation technique helps to increase the detection procedure of ripe and unripe strawberries toward autonomous strawberry harvesting robot.
Keywords
deep learning, Yolov4, data augmentation, strawberry ripeness detection, harvesting robot
References
- [1] N. Zhu et al., “Deep learning for smart agriculture: Concepts, tools, applications, and opportunities,” Int. J. Agric. Biol. Eng., vol. 11, no. 4, pp. 32–44, 2018, doi: 10.25165/j.ijabe.20181104.4475.
- [2] Y. Xiong, Y. Ge, L. Grimstad, and P. J. From, “An autonomous strawberry-harvesting robot: Design, development, integration, and field evaluation,” J. F. Robot., vol. 37, no. 2, pp. 202–224, 2020, doi: 10.1002/rob.21889.
- [3] A. De Preter, J. Anthonis, and J. De Baerdemaeker, “Development of a Robot for Harvesting Strawberries,” IFAC-PapersOnLine, vol. 51, no. 17, pp. 14–19, 2018, doi: https://doi.org/10.1016/j.ifacol.2018.08.054.
- [4] Q. Feng, J. Chen, M. Zhang, and X. Wang, “Design and Test of Harvesting Robot for Table-top Cultivated Strawberry,” in 2019 WRC Symposium on Advanced Robotics and Automation (WRC SARA), 2019, pp. 80–85, doi: 10.1109/WRC-SARA.2019.8931922.
- [5] C. W. Bac, E. J. van Henten, J. Hemming, and Y. Edan, “Harvesting robots for high‐value crops: State‐of‐the‐art review and challenges ahead,” J. F. Robot., vol. 31, no. 6, pp. 888–911, 2014.
- [6] D. Surya Prabha and J. Satheesh Kumar, “Assessment of banana fruit maturity by image processing technique,” J. Food Sci. Technol., vol. 52, no. 3, pp. 1316–1327, 2015, doi: 10.1007/s13197-013-1188-3.
- [7] W. Castro, J. Oblitas, M. De-La-Torre, C. Cotrina, K. Bazán, and H. Avila-George, “Classification of Cape Gooseberry Fruit According to its Level of Ripeness Using Machine Learning Techniques and Different Color Spaces,” IEEE Access, vol. 7, pp. 27389–27400, 2019, doi: 10.1109/ACCESS.2019.2898223.
- [8] N. El-Bendary, E. El Hariri, A. E. Hassanien, and A. Badr, “Using machine learning techniques for evaluating tomato ripeness,” Expert Syst. Appl., vol. 42, no. 4, pp. 1892–1905, 2015, doi: https://doi.org/10.1016/j.eswa.2014.09.057.
- [9] N. Goel and P. Sehgal, “Fuzzy classification of pre-harvest tomatoes for ripeness estimation – An approach based on automatic rule learning using decision tree,” Appl. Soft Comput., vol. 36, pp. 45–56, 2015, doi: https://doi.org/10.1016/j.asoc.2015.07.009.
- [10] X. Wei, K. Jia, J. Lan, Y. Li, Y. Zeng, and C. Wang, “Automatic method of fruit object extraction under complex agricultural background for vision system of fruit picking robot,” Optik (Stuttg)., vol. 125, no. 19, pp. 5684–5689, 2014, doi: https://doi.org/10.1016/j.ijleo.2014.07.001.