parent
d5768ebc89
commit
f715c740bf
File diff suppressed because it is too large
Load Diff
File diff suppressed because it is too large
Load Diff
@ -0,0 +1,98 @@
|
||||
from paddle.trainer_config_helpers import *
|
||||
|
||||
settings(batch_size=1000, learning_rate=1e-5)
|
||||
|
||||
num_channels = 3
|
||||
filter_size = 3
|
||||
filter_size_y = 3
|
||||
filter_size_z = 3
|
||||
stride = 2
|
||||
stride_y = 2
|
||||
stride_z = 2
|
||||
padding = 1
|
||||
padding_y = 1
|
||||
padding_z = 1
|
||||
groups = 1
|
||||
|
||||
data = data_layer(
|
||||
name='data1', size=12096 * num_channels, height=48, width=42, depth=6)
|
||||
|
||||
conv3d = img_conv3d_layer(
|
||||
input=data,
|
||||
name='conv3d_1',
|
||||
num_filters=16,
|
||||
num_channels=num_channels,
|
||||
filter_size=filter_size,
|
||||
filter_size_y=filter_size,
|
||||
filter_size_z=filter_size,
|
||||
stride=stride,
|
||||
stride_y=stride_y,
|
||||
stride_z=stride_z,
|
||||
padding=padding,
|
||||
padding_y=padding_y,
|
||||
padding_z=padding_z,
|
||||
groups=groups,
|
||||
bias_attr=True,
|
||||
shared_biases=True,
|
||||
trans=False,
|
||||
layer_type="conv3d",
|
||||
act=LinearActivation())
|
||||
|
||||
deconv3d = img_conv3d_layer(
|
||||
input=data,
|
||||
name='deconv3d_1',
|
||||
num_filters=16,
|
||||
num_channels=num_channels,
|
||||
filter_size=filter_size,
|
||||
filter_size_y=filter_size,
|
||||
filter_size_z=filter_size,
|
||||
stride=stride,
|
||||
stride_y=stride_y,
|
||||
stride_z=stride_z,
|
||||
padding=padding,
|
||||
padding_y=padding_y,
|
||||
padding_z=padding_z,
|
||||
groups=groups,
|
||||
bias_attr=True,
|
||||
shared_biases=True,
|
||||
trans=True,
|
||||
layer_type="deconv3d",
|
||||
act=LinearActivation())
|
||||
|
||||
data = data_layer(name="input", size=8 * 16 * 16)
|
||||
conv1 = img_conv_layer(
|
||||
input=data,
|
||||
filter_size=1,
|
||||
filter_size_y=1,
|
||||
num_channels=8,
|
||||
num_filters=16,
|
||||
stride=1,
|
||||
bias_attr=False,
|
||||
act=ReluActivation(),
|
||||
layer_type="exconv")
|
||||
conv2 = img_conv_layer(
|
||||
input=data,
|
||||
filter_size=1,
|
||||
filter_size_y=1,
|
||||
num_channels=8,
|
||||
num_filters=16,
|
||||
stride=1,
|
||||
bias_attr=False,
|
||||
act=ReluActivation(),
|
||||
layer_type="exconv")
|
||||
|
||||
concat = concat_layer(input=[conv1, conv2])
|
||||
|
||||
conv = img_conv_layer(
|
||||
input=data,
|
||||
filter_size=1,
|
||||
filter_size_y=1,
|
||||
num_channels=8,
|
||||
num_filters=16,
|
||||
stride=1,
|
||||
bias_attr=True,
|
||||
act=LinearActivation(),
|
||||
groups=2,
|
||||
layer_type="exconv")
|
||||
|
||||
outputs(concat, conv)
|
Loading…
Reference in new issue