wonkyulee9 / GlobalAttention

Global attention model for image classification

GlobalAttention

Global attention model for image classification

This repo is code for tests on global attention models for image classification. The model tries to adapt a transformer style architecture with the attention layers adapted for 2d feature maps.