roytseng-tw / Detectron.pytorch

A pytorch implementation of Detectron. Both training from scratch and inferring directly from pretrained Detectron weights are available.
MIT License
2.82k stars 567 forks source link

Curious about the Mask Loss #163

Open karenyun opened 5 years ago

karenyun commented 5 years ago

@roytseng-tw, hi, thanks for your excellent work, I just curious about the mask loss that in your implementation. In the paper, it says that they use the predicted label by the classification to select the output mask, so when computing the loss, whether it should use the target label of this rois to select the predicted mask first, and then compute the binary loss, while in your code where hasn't the step of selecting and do binary loss directly. I wonder whether my understanding is right or not, could you give some appreciate advice?