Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Fine-tuning CNN Image Retrieval with No Human Annotation

About

Image descriptors based on activations of Convolutional Neural Networks (CNNs) have become dominant in image retrieval due to their discriminative power, compactness of representation, and search efficiency. Training of CNNs, either from scratch or fine-tuning, requires a large amount of annotated data, where a high quality of annotation is often crucial. In this work, we propose to fine-tune CNNs for image retrieval on a large collection of unordered images in a fully automated manner. Reconstructed 3D models obtained by the state-of-the-art retrieval and structure-from-motion methods guide the selection of the training data. We show that both hard-positive and hard-negative examples, selected by exploiting the geometry and the camera positions available from the 3D models, enhance the performance of particular-object retrieval. CNN descriptor whitening discriminatively learned from the same training data outperforms commonly used PCA whitening. We propose a novel trainable Generalized-Mean (GeM) pooling layer that generalizes max and average pooling and show that it boosts retrieval performance. Applying the proposed method to the VGG network achieves state-of-the-art performance on the standard benchmarks: Oxford Buildings, Paris, and Holidays datasets.

Filip Radenovi\'c, Giorgos Tolias, Ond\v{r}ej Chum• 2017

Related benchmarks

TaskDatasetResultRank
Person Re-IdentificationMarket1501 (test)
Rank-1 Accuracy96.08
1264
ClassificationImageNet-1K 1.0 (val)
Top-1 Accuracy (%)82.9
1155
Visual Place RecognitionMSLS (val)
Recall@178.2
236
Vehicle Re-identificationVeRi-776 (test)
Rank-189.72
232
Visual Place RecognitionPitts30k
Recall@177.9
164
Visual Place RecognitionTokyo24/7
Recall@146.4
146
Visual Place RecognitionMSLS Challenge
Recall@149.7
134
Image RetrievalRevisited Oxford (ROxf) (Medium)
mAP76.6
124
Image RetrievalRevisited Paris (RPar) (Hard)
mAP73.2
115
Visual Place RecognitionNordland
Recall@121.6
112
Showing 10 of 70 rows

Other info

Code

Follow for update