/PyTorch-Image-Retrieval

A PyTorch framework for an image retrieval task including implementation of N-pair Loss (NIPS 2016) and Angular Loss (CVPR 2017).

Primary LanguagePythonMIT LicenseMIT

Watchers