Unet

Este archivo contiene funciones para el cálculo del coeficiente Dice como métrica y como función de costo, y para la construcción de modelos de red neuronal U-Net.

class preprocessing.unet.Unet[source]

Bases: object

static __new__(self) tensorflow.keras.models.Model[source]

Implementación del modelo de red neuronal U-Net.

Returns:

Modelo de red neuronal U-Net.

Return type:

model

dice_coef(y_true: numpy.ndarray, y_pred: numpy.ndarray) float[source]

Cálculo del coeficiente Dice como métrica.

Parameters:
  • y_true – Etiquetas.

  • y_pred – Predicciones.

Returns:

Coeficiente dice.

Return type:

float

dice_coef_loss(y_true: numpy.ndarray, y_pred: numpy.ndarray) float[source]

Cálculo del coeficiente Dice como funcion de costo.

Parameters:
  • y_true – Etiquetas.

  • y_pred – Predicciones.

Returns:

Coeficiente dice.

Return type:

float