search

nashory / DeLF-pytorch

PyTorch Implementation of "Large-Scale Image Retrieval with Attentive Deep Local Features"
MIT License
343 stars 63 forks source link

Original DELF model by authors has BN for spatial attention #36

Open dgominski opened 3 years ago

dgominski commented 3 years ago

See here the definition of the model in Tensorflow by authors : https://github.com/tensorflow/models/blob/master/research/delf/delf/python/training/model/delf_model.py

They use BatchNorm between conv1 and conv2 of the spatial attention module.