search

jshtok / RepMet

Few-shot detection for visual categories
Apache License 2.0
108 stars 18 forks source link

RPN classification Loss #18

Closed yang-yk closed 4 years ago

yang-yk commented 4 years ago

https://github.com/jshtok/RepMet/blob/22aec096cd89a06d008fdd5b335a816545bdf073/fpn/symbols/resnet_v1_101_fpn_dcn_rcnn_oneshot_v3.py#L953

Sorry to bother you. I wonder why there is no make_loss for RPN classification loss? @jshtok @duynn912

jshtok commented 4 years ago

Hi @yang-yk, In MXNet the SoftmaxOutput is already a loss, so there is no need for MakeLoss.

yang-yk commented 4 years ago

Thanks!