KevinMusgrave/pytorch-metric-learning

Train and evaluate embeddings for similarity search with PyTorch. Includes contrastive losses, hard example mining, samplers, and retrieval metrics.

语言：Python · GitHub 星标：6,342 · Fork：663

话题：computer-vision · contrastive-learning · deep-learning · deep-metric-learning · embeddings · image-retrieval · machine-learning · metric-learning

在 GitHub 上查看原仓库