Share your thoughts, 1 month free Claude Pro on usSee more
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
1163
Person Re-IdentificationMarket 1501
mAP92
1071
Person Re-IdentificationMSMT17
mAP0.559
514
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
Showing 10 of 76 rows
...

Other info

Code

Follow for update