|
|
@ -171,7 +171,7 @@ class SE_ResNeXt():
|
|
|
|
num_filters=num_filters,
|
|
|
|
num_filters=num_filters,
|
|
|
|
filter_size=filter_size,
|
|
|
|
filter_size=filter_size,
|
|
|
|
stride=stride,
|
|
|
|
stride=stride,
|
|
|
|
padding=(filter_size - 1) / 2,
|
|
|
|
padding=(filter_size - 1) // 2,
|
|
|
|
groups=groups,
|
|
|
|
groups=groups,
|
|
|
|
act=None,
|
|
|
|
act=None,
|
|
|
|
bias_attr=False)
|
|
|
|
bias_attr=False)
|
|
|
@ -182,7 +182,7 @@ class SE_ResNeXt():
|
|
|
|
input=input, pool_size=0, pool_type='avg', global_pooling=True)
|
|
|
|
input=input, pool_size=0, pool_type='avg', global_pooling=True)
|
|
|
|
stdv = 1.0 / math.sqrt(pool.shape[1] * 1.0)
|
|
|
|
stdv = 1.0 / math.sqrt(pool.shape[1] * 1.0)
|
|
|
|
squeeze = fluid.layers.fc(input=pool,
|
|
|
|
squeeze = fluid.layers.fc(input=pool,
|
|
|
|
size=num_channels / reduction_ratio,
|
|
|
|
size=num_channels // reduction_ratio,
|
|
|
|
act='relu')
|
|
|
|
act='relu')
|
|
|
|
stdv = 1.0 / math.sqrt(squeeze.shape[1] * 1.0)
|
|
|
|
stdv = 1.0 / math.sqrt(squeeze.shape[1] * 1.0)
|
|
|
|
excitation = fluid.layers.fc(input=squeeze,
|
|
|
|
excitation = fluid.layers.fc(input=squeeze,
|
|
|
|