KevinMusgrave / pytorch-metric-learning

The easiest way to use deep metric learning in your application. Modular, flexible, and extensible. Written in PyTorch.
https://kevinmusgrave.github.io/pytorch-metric-learning/
MIT License
6.01k stars 658 forks source link

How to train binary descriptor in image retrieval use this project #493

Closed ghoshaw closed 2 years ago

ghoshaw commented 2 years ago

HI, could you give some example for how to use the loss in this project to train binary descriptor in image retrieval ? like in this paper 'Simultaneous Feature Learning and Hash Coding with Deep Neural Networks'. Thanks!

KevinMusgrave commented 2 years ago

I don't think this library will work for that. Check out these libraries instead:

https://github.com/swuxyj/DeepHash-pytorch https://github.com/thulab/DeepHash

ghoshaw commented 2 years ago

ok, thanks