fyang93 / diffusion

Efficient Diffusion for Image Retrieval
MIT License
221 stars 36 forks source link

Where does the feature of a query come from? #4

Closed gdjmck closed 5 years ago

gdjmck commented 5 years ago

In the paper, the input feature is 512 or 1024 dimension R-MAC descriptor, but in the downloaded data it is 2048 dimension, just like the last feature in resnet101. Another question is that is resnet model just pretrained on ImageNet or other training data are used? Thank you!

fyang93 commented 5 years ago

@gdjmck, that could be a mistake. As far as I know, ResNet50 produces 2048-d descriptors at the last convolution layer. As for the model, you may refer to this paper for further details, we only used their extracted descriptors.