qubvel / segmentation_models

Segmentation models with pretrained backbones. Keras and TensorFlow Keras.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

How to implement more metrics in model.compile?

nora-dimitrova opened this issue · comments

I want to evaluate segmentation results with Dice, Jaccard, True positive - False positive and True positive - False negative ratio. I saw the metrics script and how to implement each of them individually but not all of them. How can I get them together, also keeping mse?

.....
model = sm.Unet(BACKBONE, encoder_weights = "imagenet")
model.compile(optimizer = "adam", loss = "binary_crossentropy", metrics = ["mse"]) #mse mean squared error
.....