Skip to content

Instantly share code, notes, and snippets.

View jrimestad's full-sized avatar

Jens Rimestad jrimestad

1 follower · 8 following
  • Vitrolife
  • Danmark
View GitHub Profile
@jrimestad
jrimestad / keras_attention_2d.py
Created February 12, 2019 01:21
A 2D attention block from https://www.kaggle.com/c/human-protein-atlas-image-classification/discussion/77269#454015
from __future__ import absolute_import, division
import sys
from os.path import dirname
sys.path.append(dirname(dirname(__file__)))
from keras import initializers
from keras.engine import InputSpec, Layer
from keras import backend as K