Clasificación del fruto del tomate utilizando Yolov5 y modelos de redes neuronales convolucionales
Autores: Phan, Quoc-Hung; Nguyen, Van-Tung; Lien, Chi-Hsiang; Duong, The-Phong; Hou, Max Ti-Kuang; Le, Ngoc-Bich
Idioma: Inglés
Editor: MDPI
Año: 2023
Acceso abierto
Artículo científico
2023
Clasificación del fruto del tomate utilizando Yolov5 y modelos de redes neuronales convolucionalesCategoría
Ciencias Agrícolas y Biológicas
Subcategoría
Botánica
Palabras clave
Marcos de aprendizaje profundo
Yolov5m
ResNet50
ResNet-101
EfficientNet-B0
Fruta de tomate
Licencia
CC BY-SA – Atribución – Compartir Igual
Consultas: 5
Citaciones: Sin citaciones
Se proponen cuatro marcos de aprendizaje profundo que consisten en Yolov5m y Yolov5m combinados con ResNet50, ResNet-101 y EfficientNet-B0, respectivamente, para clasificar los frutos de tomate en la planta en tres categorías: maduro, inmaduro y dañado. Para un conjunto de datos de entrenamiento que consiste en 4500 imágenes y un proceso de entrenamiento con 200 épocas, un tamaño de lote de 128 y un tamaño de imagen de 224 x 224 píxeles, se encuentra que la precisión de predicción para tomates maduros e inmaduros es del 100% al combinar Yolo5m con ResNet-101. Mientras tanto, la precisión de predicción para tomates dañados es del 94% al usar Yolo5m con el modelo Efficient-B0. Las redes ResNet-50, EfficientNet-B0, Yolov5m y ResNet-101 tienen precisiones de prueba del 98%, 98%, 97% y 97%, respectivamente. Por lo tanto, los cuatro marcos tienen el potencial para la clasificación de frutos de tomate en aplicaciones de cosecha automatizada de tomates en la agricultura.
Descripción
Se proponen cuatro marcos de aprendizaje profundo que consisten en Yolov5m y Yolov5m combinados con ResNet50, ResNet-101 y EfficientNet-B0, respectivamente, para clasificar los frutos de tomate en la planta en tres categorías: maduro, inmaduro y dañado. Para un conjunto de datos de entrenamiento que consiste en 4500 imágenes y un proceso de entrenamiento con 200 épocas, un tamaño de lote de 128 y un tamaño de imagen de 224 x 224 píxeles, se encuentra que la precisión de predicción para tomates maduros e inmaduros es del 100% al combinar Yolo5m con ResNet-101. Mientras tanto, la precisión de predicción para tomates dañados es del 94% al usar Yolo5m con el modelo Efficient-B0. Las redes ResNet-50, EfficientNet-B0, Yolov5m y ResNet-101 tienen precisiones de prueba del 98%, 98%, 97% y 97%, respectivamente. Por lo tanto, los cuatro marcos tienen el potencial para la clasificación de frutos de tomate en aplicaciones de cosecha automatizada de tomates en la agricultura.