HyperAIHyperAI

Command Palette

Search for a command to run...

4 months ago

Fine-tuning CNN Image Retrieval with No Human Annotation

Filip Radenović; Giorgos Tolias; Ondřej Chum

Fine-tuning CNN Image Retrieval with No Human Annotation

Abstract

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.

Code Repositories

filipradenovic/cnnimageretrieval
pytorch
Mentioned in GitHub
raojay7/cnnimageretrieval-pytorch
pytorch
Mentioned in GitHub
naver/deep-image-retrieval
pytorch
Mentioned in GitHub
almazan/deep-image-retrieval
pytorch
Mentioned in GitHub
osimeoni/DSM
Mentioned in GitHub
nikosefth/freedom
pytorch
Mentioned in GitHub
layer6ai-labs/GSS
tf
Mentioned in GitHub
RuibinMa/comp755project-ruibinma
pytorch
Mentioned in GitHub
filipradenovic/cnnimageretrieval-pytorch
Official
pytorch
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
image-retrieval-on-roxford-hardR–GeM
mAP: 38.5
image-retrieval-on-roxford-mediumR–GeM
mAP: 64.7
image-retrieval-on-rparis-hardR–GeM
mAP: 56.3
image-retrieval-on-rparis-mediumR–GeM
mAP: 77.2

Build AI with AI

From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.

AI Co-coding
Ready-to-use GPUs
Best Pricing
Get Started

Hyper Newsletters

Subscribe to our latest updates
We will deliver the latest updates of the week to your inbox at nine o'clock every Monday morning
Powered by MailChimp
Fine-tuning CNN Image Retrieval with No Human Annotation | Papers | HyperAI