← all repositories

filipradenovic/cnnimageretrieval-pytorch

A PyTorch toolbox for training CNNs and learning supervised whitening for image retrieval on Oxford and Paris benchmark datasets.

1.5k stars Python Computer VisionML Frameworks
cnnimageretrieval-pytorch
Velocity · 7d
+0.5
★ / day
Trend
steady
star history

This repository provides a complete pipeline for CNN-based image retrieval, including fine-tuning pre-trained convolutional neural networks with metric learning losses, learning supervised whitening as post-processing for global descriptors, and evaluating on standard benchmarks. The approach handles hard example mining during training and supports various architectures, pooling methods, and whitening strategies. The code builds on the PyTorch framework for tensor operations and automatic differentiation.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.