huyhoang17 / DOLG-pytorch

Unofficial PyTorch Implementation of "DOLG: Single-Stage Image Retrieval with Deep Orthogonal Fusion of Local and Global Features"

Pytorch Implementation of Deep Orthogonal Fusion of Local and Global Features (DOLG)

This is the unofficial PyTorch Implementation of "DOLG: Single-Stage Image Retrieval with Deep Orthogonal Fusion of Local and Global Features"

reference: https://arxiv.org/pdf/2108.02927.pdf

Model Structure

Prerequisites

• PyTorch
• PyTorch Lightning
• timm
• sklearn
• pandas
• jpeg4py
• albumentations
• python3
• CUDA

Data

You can get the GLDv2 dataset from here.

If you just want the GLDv2-clean dataset, check this kaggle competition dataset.

Place your data like the structure below

data
├── train_clean.csv
└── train
    └── ###
        └── ###
            └── ###
                └── ###.jpg

Citations

@misc{yang2021dolg,
      title={DOLG: Single-Stage Image Retrieval with Deep Orthogonal Fusion of Local and Global Features}, 
      author={Min Yang and Dongliang He and Miao Fan and Baorong Shi and Xuetong Xue and Fu Li and Errui Ding and Jizhou Huang},
      year={2021},
      eprint={2108.02927},
      archivePrefix={arXiv},
      primaryClass={cs.CV}
}