panel_segmentation.panel_train.TrainPanelSegmentationModel.diceCoeffLoss¶
-
TrainPanelSegmentationModel.
diceCoeffLoss
(y_true, y_pred)[source]¶ This function is a loss function that can be used when training the segmentation model. This loss function can be used in place of binary crossentropy, which is the current loss function in the training stage.
- Parameters
y_true (
nparray float
) – The true mask of the imagey_pred (
nparray float
) – The predicted mask of the data
- Returns
float
The loss metric between prediction and ground truth