loss for predicted binary coding

[hrkita]

Hi，

If I want to compare the entire mask to obtain a loss function, what part of the code do I need to modify? Or do I need to retrain the ResNet34 model?

if predict_entire_mask: loss_m = maskLoss(pred_mask_prob, entire_masks) else: loss_m = maskLoss(pred_mask_prob, masks)

Best, Joa

[suyz526]

If you set this arg as True, the network will predict entire mask instead of visible mask.

[hrkita]

Okay, I will give it a try.Thanks!