A PyTorch framework for an image retrieval task including implementation of N-pair Loss (NIPS 2016) and Angular Loss (ICCV 2017).
A PyTorch framework for an image retrieval task including implementation of N-pair Loss (NIPS 2016) and Angular Loss (ICCV 2017).
We implemented loss functions to train the network for image retrieval.
Batch sampler for the loss function borrowed from here.
We attached the self-attention module of the Self-Attention GAN to conventional classification networks (e.g. DenseNet, ResNet, or SENet).
Implementation of the module borrowed from here.
We adopted data augmentation techniques used in Single Shot MultiBox Detector.
We utilized the following post-processing techniques in the inference phase.