如何实施'组'张量层中的alexnet

时间:2017-04-24 12:52:57

标签: tensorflow tflearn

group用于将卷积内核(将前一层和当前层连接)的参数分组为强制在alexnet中的k部分,是否有一个简单的工具用于张量层中的组?

1 个答案:

答案 0 :(得分:1)

This可能是一个有用的链接。您需要在卷积之前拆分conv层,然后连接结果。

如果有帮助的话,我们会解释如何组织Caffe提供的模型BVLC's alexnet的权重here

您需要将caffemodel转换为张量流可读格式(根据我的经验)。转换为.npy文件的权重提供here(bvlc_alexnet.npy),但将其转换为您选择的格式应该很简单(例如,.h5)。