Skip to content

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

License

Notifications You must be signed in to change notification settings

dongkyunk/DOLG-pytorch

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

9 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

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

Image

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}
}

About

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

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages