2020/08/20 by Kevin Musgrave, Serge Belongie, Musgrave, Kevin +3 · 3 citations
Computer Science · #Face and Expression Recognition #Machine Learning and Data Classification #Image Retrieval and Classification Techniques
paper · pdf · doi:10.48550/arxiv.2008.09164
Deep metric learning algorithms have a wide variety of applications, but implementing these algorithms can be tedious and time consuming. PyTorch Metric Learning is an open source library that aims to remove this barrier for both researchers and practitioners. The modular and flexible design allows users to easily try out different combinations of algorithms in their existing code. It also comes with complete train/test workflows, for users who want results fast. Code and documentation is available at https://www.github.com/KevinMusgrave/pytorch-metric-learning.