/solo

Primary LanguageC++

Instance Segmentation

my libtorch implementation for SOLO: Segementing Objects By Locations

  1. https://www.zhihu.com/question/360594484/answer/936591301
  2. https://towardsdatascience.com/single-stage-instance-segmentation-a-review-1eeb66e0cc49

Dice Loss: for mask prediction

https://stats.stackexchange.com/questions/438494/what-is-the-intuition-behind-what-makes-dice-coefficient-handle-imbalanced-data

Focal Loss: for classification

https://zhuanlan.zhihu.com/p/49981234

BCELoss, BCEWithLogits

https://medium.com/@zhang_yang/how-is-pytorchs-binary-cross-entropy-with-logits-function-related-to-sigmoid-and-d3bd8fb080e7 BCEWithLogits == sigmoid+BCE