中文

PyTorch Metric Learning

计算机视觉与模式识别 2020-08-24 v1 机器学习

摘要

深度度量学习算法有广泛的应用,但实现这些算法可能繁琐且耗时。PyTorch Metric Learning 是一个开源库,旨在为研究者和从业者消除这一障碍。其模块化与灵活的设计使用户能在现有代码中轻松尝试不同的算法组合。它还带有完整的训练/测试工作流,供希望快速获得结果的用户使用。代码与文档可在 https://www.github.com/KevinMusgrave/pytorch-metric-learning 获取。

关键词

引用

@article{arxiv.2008.09164,
  title  = {PyTorch Metric Learning},
  author = {Kevin Musgrave and Serge Belongie and Ser-Nam Lim},
  journal= {arXiv preprint arXiv:2008.09164},
  year   = {2020}
}

备注

Code and documentation is available at https://www.github.com/KevinMusgrave/pytorch-metric-learning