Extracting enhanced image features

[MLRadfys]

Hi and thanks for providing you're research :-)

Iam interested in extracting the "updated image features" after the feature enhancer layer. Unfortunately I wasn't able to find the correct spot in the code base, or at least Iam unsure.

Is it correct that the variable v in the BiAttentionBlock corresponds to the enhanced features?

I would be really happy if you would be able to give me some hint.

Kind regards,

M