Embedding of individual patches

[Sino0904]

Hello,

This is really a great work! thanks a lot for making the codebase public. I was wondering if its possible to get individual embedding of each patch of size (q x q). From the pytorch model architecture, I was not sure how to do that.

Thanks!

[wielandbrendel]

@Sino0904 You could simply modify the output of the Bagnet here:

https://github.com/wielandbrendel/bag-of-local-features-models/blob/master/bagnets/pytorchnet.py#L110-L129

where you would have to leave out the fc-layer. One could even make this an argument so you can choose whether you get logits or embeddings returned. Feel free to open a PR if you implement that.