I’m retraining resnet101 to work better on my dataset for image segmentation. I’m not sure which metric to use to calculate the loss. Can someone help?
BBloggsbott (Vishal R) #1
You say metric and also loss, these are two separate things.
Good metrics include IOU and Dice.
BBloggsbott (Vishal R) #3
Oh, Okay. I’ll look into the difference. What will be a good loss function?
fastai uses by default pixel-wise cross entropy loss.
BBloggsbott (Vishal R) #5
Okay. Thank you.