The 1st Place Solution of the Google Landmark 2019 Retrieval Challenge and the 3rd Place Solution of the Recognition Challenge.
We have published two papers regarding our solution. You can check from:
- Large-scale Landmark Retrieval/Recognition under a Noisy and Diverse Dataset
- Tech report describing our solution (4-pages)
- Two-stage Discriminative Re-ranking for Large-scale Landmark Retrieval
- Extended version of above tech report, including more detailed explanation and additional experimental results (10-pages, CVPR Workshop 2020)
You can reproduce our environments using Dockerfile provided here https://github.com/lyakaap/Landmark2019-1st-and-3rd-Place-Solution/blob/master/docker/Dockerfile
- Google Landmark Dataset v1 (GLD-v1): https://www.kaggle.com/google/google-landmarks-dataset
- Google Landmark Dataset v2 (GLD-v2): https://github.com/cvdfoundation/google-landmark
- Clean version of the v2 (Newly Released!): https://www.kaggle.com/confirm/cleaned-subsets-of-google-landmarks-v2/kernels
Dataset statistics:
| Dataset (train split) | # Samples | # Labels |
|---|---|---|
| GLD-v1 | 1,225,029 | 14,951 |
| GLD-v2 | 4,132,914 | 203,094 |
| GLD-v2 (clean) | 1,580,470 | 81,313 |
(You can skip this procedure to generate a cleaned subset. Pre-computed files are available on kaggle dataset.)
Run scripts/prepare_cleaned_subset.sh for cleaning the GLD-v2 dataset.
The cleaning code requires DELF library (install instructions).
Model training and inference are done in exp/ directory.
# train models by various parameter settings with 4 gpus (each training is done with 2 gpus).
python vX.py tuning -d 0,1,2,3 --n-gpu 2
# predict
python vX.py predict -m vX/epX.pth -d 0
# predict with multiple gpus
python vX.py multigpu-predict -m vX/epX.pth --scale L2 --ms -b 32 -d 0,1
| Place | Team | Private | Public |
|---|---|---|---|
| 1st | smlyaka (ours) | 37.23 | 35.69 |
| 2nd | imagesearch | 34.75 | 32.25 |
| 3rd | Layer 6 AI | 32.18 | 29.85 |
- https://github.com/filipradenovic/cnnimageretrieval-pytorch/tree/master/cirtorch
- https://github.com/ronghuaiyang/arcface-pytorch/blob/master/models/metrics.py
- https://github.com/4uiiurz1/pytorch-adacos/blob/master/metrics.py
- https://github.com/kevin-ssy/FishNet
- https://github.com/facebookresearch/maskrcnn-benchmark/blob/master/maskrcnn_benchmark/data/samplers/grouped_batch_sampler.py
