search

KaiyangZhou / deep-person-reid

Torchreid: Deep learning person re-identification in PyTorch.
https://kaiyangzhou.github.io/deep-person-reid/
MIT License
4.25k stars 1.14k forks source link

How to modify softmax #342

Open szxczyc opened 4 years ago

szxczyc commented 4 years ago

hi, kaiyang: If I want to change softmax to AMsoftmax, etc., what should I do? Is it only need to modify torchreid / engine / image / softmax.py?

KaiyangZhou commented 4 years ago

you can add the AMSoftmax loss in torchreid/losses

maybe this could help https://github.com/opencv/openvino_training_extensions/tree/develop/pytorch_toolkit/person_reidentification

szxczyc commented 4 years ago

Thank you very much!

CompliceU commented 4 years ago

@szxczyc I don't know how the effect of this loss function is now

Scorpio-kk commented 1 year ago

嗨，开阳：如果我想将softmax更改为AMsoftmax等，我该怎么办？只需要修改吗？torchreid / engine / image / softmax.py

Have you succeeded? Can you share your code? I need help