kevinmusgrave/pytorch-metric-learning: The easiest way to use deep metric learning in your application. Modular, flexible, and extensible. Written in PyTorch.
Category: ml
Language: Python
Topics: computer-vision, contrastive-learning, deep-learning, deep-metric-learning, embeddings, image-retrieval, machine-learning, metric-learning, pytorch, self-supervised-learning
Homepage: https://kevinmusgrave.github.io/pytorch-metric-learning/