在pytorch中没有找到Inception模块,自己写了一个,以供调用。
Inception模块的顺序为:
1. 输入 -> 1*1卷积 -> BatchNorm -> ReLU -> 1*5卷积 -> BatchNorm -> ReLU
2. 输入 -> 1*1卷积 -> BatchNorm -> ReLU -> 1*3卷积 -> BatchNorm -> ReLU
3. 输入 -> 池化 -> 1*1卷积 -> BatchNorm -> ReLU
4. 输入 -> 1*1卷积 -> BatchNorm -> ReLU
其中,1和2步骤可以重复多次。最后将所有结果串接起来。
pytorch中实现如下,应用例子见我的下一篇文章:openface(三):卷积网络。
import torch.nn as nn
class Inception(nn.Module):
def __init__(self, inputSize, kernelSize, kernelStride, outputSize,