Learning local feature descriptors with triplets and shallow convolutional neural networks

Code

• github.com/vbalnt/tfeat
  In paperpytorch★ 0
• github.com/kornia/kornia/blob/master/kornia/feature/tfeat.py
  pytorch★ 0

Abstract

It has recently been demonstrated that local feature descriptors based on convolutional neural networks (CNN) can significantly improve the matching performance. Previous work on learning such descriptors has focused on exploiting pairs of positive and negative patches to learn discriminative CNN representations. In this work, we propose to utilize triplets of training samples, together with in-triplet mining of hard negatives. We show that our method achieves state of the art results, without the computational overhead typically associated with mining of negatives and with lower complexity of the network architecture. We compare our approach to recently introduced convolutional local feature descriptors, and demonstrate the advantages of the proposed methods in terms of performance and speed. We also examine different loss functions associated with triplets.

Tasks

Reproductions