Analizar la salida del modelo

Después de obtener la predicción de un modelo preentrenado, ahora puedes analizarla en detalle. prediction está disponible para ti.

Verás que es una lista de longitud uno (hiciste la predicción para una sola imagen) con un diccionario con cuatro claves. labels son enteros que indican las clases predichas y scores son las probabilidades correspondientes. Puedes acceder a la lista de nombres de clase correspondientes a labels como class_names.

¿Cuáles son las categorías predichas y sus puntuaciones asociadas para las dos máscaras principales?