项目作者: johnsmithm

项目描述 :
Multi heads attention for image classification
高级语言: Python
项目地址: git://github.com/johnsmithm/multi-heads-attention-image-classification.git


Attention is all you need: A Keras Implementation

Using attention to increase image classification accuracy. Inspired from “Attention is All You Need“ (Ashish Vaswani, Noam Shazeer, Niki Parmar, Jakob Uszkoreit, Llion Jones, Aidan N. Gomez, Lukasz Kaiser, Illia Polosukhin, arxiv, 2017).

The medium article can be found @moshnoi2000/all-you-need-is-attention-computer-vision-edition-dbe7538330a4">here.