PyTorch Image Retrieval
We implemented loss functions to train the network for image retrieval.
Batch sampler for the loss function borrowed from here.
- N-pair Loss (NIPS 2016): Sohn, Kihyuk. "Improved Deep Metric Learning with Multi-class N-pair Loss Objective," Advances in Neural Information Processing Systems. 2016.
- Angular Loss (CVPR 2017): Wang, Jian. "Deep Metric Learning with Angular Loss," CVPR, 2017
We adopted data augmentation techniques used in Single Shot MultiBox Detector.
We utilized the following post-processing techniques in the inference phase.