search

ZQPei / deep_sort_pytorch

MOT using deepsort and yolov3 with pytorch
MIT License
2.83k stars 727 forks source link

softmax vs cosine softmax ? #257

Open YoungjaeDev opened 1 year ago

YoungjaeDev commented 1 year ago

Hi In the official code (https://github.com/nwojke/deep_sort), he focused on generative learning using cosine_softmax loss image

However, in the code implemented by pytorch that code refer to, it simply conducted a discrimitive learning with cross_entropy

https://github.com/ZQPei/deep_sort_pytorch/blob/8cfe2467a4b1f4421ebf9fcbc157921144ffe7cf/deep_sort/deep/train.py#L67

Is there a big difference between the reason and the performance of this implementation?

yesmineht commented 1 year ago

hello @youngjae-avikus did you change it to cosine softmax_loss ?