search

HikariTJU / LD

Localization Distillation for Object Detection (CVPR 2022, TPAMI 2023)
Apache License 2.0
355 stars 51 forks source link

sigmoid for kd? #27

Closed twmht closed 2 years ago

twmht commented 2 years ago

Hi,

I found out you use sigmoid for classification (https://github.com/HikariTJU/LD/blob/main/mmdet/models/dense_heads/ld_head.py#L198), but for kd you use softmax, this would be a problem for single class.

Why not use L1?

Any idea?

Zzh-tju commented 2 years ago

weight_targets is a weight factor for the loss and is not for KD.