¿Cómo obtener la matriz de confusión del modelo YOLOv4-tiny personalizado entrenado?
Después de entrenar un modelo Yolov4-tiny con datos personalizados, ¿cómo puedo obtener la matriz de confusión para mis pesos entrenados?
Si tienes alguna pregunta, puedes hacerla a continuación o ingresar lo que estás buscando.
davy.ai
Para obtener una matriz de confusión para tu modelo Yolov4-tiny entrenado con datos personalizados, puedes seguir los siguientes pasos:
Utiliza la función
detect_dataset
en el archivodetect.py
para obtener las predicciones para cada imagen en el conjunto de datos de prueba. Esto dará como resultado una lista de diccionarios, en donde cada diccionario contiene las cajas delimitadoras predichas y sus etiquetas de clase correspondientes para una imagen.Convierte las cajas delimitadoras predichas a máscaras binarias, para que puedas compararlas con las máscaras de verdad. Para hacer esto, puedes utilizar el módulo
visualize_utils
.Carga las anotaciones de verdad para el conjunto de datos de prueba y conviértelas también en máscaras binarias.
Utiliza las máscaras binarias para calcular la matriz de confusión para el modelo. Para hacer esto, puedes utilizar el módulo
sklearn.metrics
. La matriz de confusión te mostrará la cantidad de predicciones verdaderas positivas, verdaderas negativas, falsas positivas y falsas negativas para cada clase, lo que te ayudará a evaluar el rendimiento de tu modelo.Finalmente, puedes visualizar la matriz de confusión utilizando la biblioteca
matplotlib
.En general, el proceso para obtener una matriz de confusión para tu modelo Yolov4-tiny entrenado involucra evaluar el modelo en un conjunto de datos de prueba, convertir las cajas delimitadoras predichas y de verdad en máscaras binarias, calcular la matriz de confusión utilizando las máscaras binarias y visualizar la matriz.