AttentionNet.py 文件源码

python
阅读 18 收藏 0 点赞 0 评论 0

项目:AttentionNet 作者: sayvazov 项目源码 文件源码
def build_classification_network(r1):
    if not isinstance(r1, lasagne.layers.Layer):
        l_in = lasagne.layers.InputLayer((None, glimpse_output_size, recurrent_output_size), r1)
    else:
        l_in = r1
    output = lasagne.layers.DenseLayer(l_in, classification_units, 
                                       nonlinearity = nl.softmax, 
                                       W = class_weights, b = class_bias)
    return output

#input is downsampled batch of images
#output is initial r2, of length glimpse_output_size
评论列表
文章目录


问题


面经


文章

微信
公众号

扫码关注公众号