antorsae / landmark-retrieval-2019

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Retrieval Challenge 2019 solution

by Andres Torrubia

The solution consists of two steps to replicate our solution (0.10439 private 0.08832 public) for https://www.kaggle.com/c/landmark-retrieval-2019:

Single model CNN extraction + NN search

This notebook extracts the last convolution layer of a given architecture applied GeM pooling and performs L2 normalization.

It uses augmentation (images are LR flipped) so both the index and queries features are duplicated.

Once features are extracted, it performs regular nearest neighbour search using PCA, whitening and L2 normalization and the resuls are ready for a submission.

It also saves the results of nearest neighbour search for ensembling.

Ensembling

This notebook takes the results of nearest neighbours search for each query (and flipped LR queries) and aggregates distances of different runs (different architectures) and builds a submission accordingly.

For flipped LR images, we pick the minimum distance.

Some results

image image image

About


Languages

Language:Jupyter Notebook 100.0%Language:Python 0.0%