tensorflow - 如何在张量层中实现 alexnet 的“组”

Question

group 用于在 alexnet 中将卷积核（连接前一层和当前层）的参数强制分组为 k 个部分，在 tensorlayer 中有一个简单的 group 实现吗？

Answer 1

这可能是一个有用的链接。您需要在卷积之前拆分卷积层，然后将结果连接起来。

如果有帮助，这里会解释如何组织 Caffe 提供的模型中BVLC的alexnet的权重。

您需要将 caffemodel 转换为 tensorflow 可读格式（根据我的经验）。此处提供了转换为 .npy 文件的权重(bvlc_alexnet.npy)，但将其转换为您选择的格式（例如 .h5）应该很简单。