why remove "background" class?

chengyangfu / retinamask

RetinaMask

MIT License

339 stars 52 forks source link

why remove "background" class? #14

Closed boundles closed 5 years ago

boundles commented 5 years ago

"box_cls" shape is num_class-1, so why remove "background" class?

chengyangfu commented 5 years ago

This follows the implementation of origianl FocalLoss paper. FocalLoss is a weighted version of multi-class sigmoid cross entropy loss function. Each class is treated independently. I also tried the softmax version but didn't get better results.

boundles commented 5 years ago

OK, got it, thanks for your reply.