def create_in_sequence(n_chans, out_filters):
return nn.Sequential(
weight_scale(nn.Conv2d(1, out_filters, (101, 3), padding=(50, 1)),
gain=calculate_gain('leaky_relu')), nn.LeakyReLU(0.2),
weight_scale(nn.Conv2d(out_filters, out_filters, (1, n_chans)),
gain=calculate_gain('leaky_relu')),
Reshape([[0], [1], [2]]), nn.LeakyReLU(0.2))
def create_gen_blocks(n_chans,z_vars):
def create_conv_sequence(in_filters,out_filters):
return nn.Sequential(CubicUpsampling1d(2),
weight_scale(nn.Conv1d(in_filters,out_filters,9,padding=4),
gain=calculate_gain('leaky_relu')),
nn.LeakyReLU(0.2),
PixelNorm(),
weight_scale(nn.Conv1d(out_filters,out_filters,9,padding=4),
gain=calculate_gain('leaky_relu')),
nn.LeakyReLU(0.2),
PixelNorm())
def create_out_sequence(n_chans,in_filters):
return nn.Sequential(weight_scale(nn.Conv1d(in_filters,n_chans,1),
gain=calculate_gain('linear')),
Reshape([[0],[1],[2],1]),
PixelShuffle2d([1,n_chans]))
def create_fade_sequence(factor):
return CubicUpsampling2d(2)
blocks = []
tmp_block = ProgressiveGeneratorBlock(
nn.Sequential(weight_scale(nn.Linear(z_vars,50*12),
gain=calculate_gain('leaky_relu')),
nn.LeakyReLU(0.2),
Reshape([[0],50,-1]),
create_conv_sequence(50,50)),
create_out_sequence(n_chans,50),
create_fade_sequence(2)
)
blocks.append(tmp_block)
tmp_block = ProgressiveGeneratorBlock(
create_conv_sequence(50,50),
create_out_sequence(n_chans,50),
create_fade_sequence(2)
)
blocks.append(tmp_block)
tmp_block = ProgressiveGeneratorBlock(
create_conv_sequence(50,50),
create_out_sequence(n_chans,50),
create_fade_sequence(2)
)
blocks.append(tmp_block)
tmp_block = ProgressiveGeneratorBlock(
create_conv_sequence(50,50),
create_out_sequence(n_chans,50),
create_fade_sequence(2)
)
blocks.append(tmp_block)
tmp_block = ProgressiveGeneratorBlock(
create_conv_sequence(50,50),
create_out_sequence(n_chans,50),
create_fade_sequence(2)
)
blocks.append(tmp_block)
tmp_block = ProgressiveGeneratorBlock(
create_conv_sequence(50,50),
create_out_sequence(n_chans,50),
None
)
blocks.append(tmp_block)
return blocks
def create_disc_blocks(n_chans):
def create_conv_sequence(in_filters,out_filters):
return nn.Sequential(weight_scale(nn.Conv1d(in_filters,in_filters,9,padding=4),
gain=calculate_gain('leaky_relu')),
nn.LeakyReLU(0.2),
weight_scale(nn.Conv1d(in_filters,out_filters,9,padding=4),
gain=calculate_gain('leaky_relu')),
nn.LeakyReLU(0.2),
weight_scale(nn.Conv1d(out_filters,out_filters,2,stride=2),
gain=calculate_gain('leaky_relu')),
nn.LeakyReLU(0.2))
def create_in_sequence(n_chans,out_filters):
return nn.Sequential(weight_scale(nn.Conv2d(1,out_filters,(1,n_chans)),
gain=calculate_gain('leaky_relu')),
Reshape([[0],[1],[2]]),
nn.LeakyReLU(0.2))
def create_fade_sequence(factor):
return nn.AvgPool2d((factor,1),stride=(factor,1))
blocks = []
tmp_block = ProgressiveDiscriminatorBlock(
create_conv_sequence(50,50),
create_in_sequence(n_chans,50),
create_fade_sequence(2)
)
blocks.append(tmp_block)
tmp_block = ProgressiveDiscriminatorBlock(
create_conv_sequence(50,50),
create_in_sequence(n_chans,50),
create_fade_sequence(2)
)
blocks.append(tmp_block)
tmp_block = ProgressiveDiscriminatorBlock(
create_conv_sequence(50,50),
create_in_sequence(n_chans,50),
create_fade_sequence(2)
)
blocks.append(tmp_block)
tmp_block = ProgressiveDiscriminatorBlock(
create_conv_sequence(50,50),
create_in_sequence(n_chans,50),
create_fade_sequence(2)
)
blocks.append(tmp_block)
tmp_block = ProgressiveDiscriminatorBlock(
create_conv_sequence(50,50),
create_in_sequence(n_chans,50),
create_fade_sequence(2)
)
blocks.append(tmp_block)
tmp_block = ProgressiveDiscriminatorBlock(
nn.Sequential(StdMap1d(),
create_conv_sequence(51,50),
Reshape([[0],-1]),
weight_scale(nn.Linear(50*12,1),
gain=calculate_gain('linear'))),
create_in_sequence(n_chans,50),
None
)
blocks.append(tmp_block)
return blocks
def create_gen_blocks(n_chans, z_vars):
def create_conv_sequence(in_filters, out_filters):
return nn.Sequential(
upsample_layer(mode='linear', scale_factor=2),
weight_scale(nn.Conv1d(in_filters,
out_filters,
21,
padding=21 // 2),
gain=calculate_gain('leaky_relu')), nn.LeakyReLU(0.2),
PixelNorm(),
weight_scale(nn.Conv1d(out_filters, out_filters, 9,
padding=9 // 2),
gain=calculate_gain('leaky_relu')), nn.LeakyReLU(0.2),
PixelNorm())
def create_out_sequence(n_chans, in_filters):
return nn.Sequential(
weight_scale(nn.Conv1d(in_filters, n_chans, 1),
gain=calculate_gain('linear')),
Reshape([[0], [1], [2], 1]), PixelShuffle2d([1, n_chans]))
def create_fade_sequence(factor):
return upsample_layer(mode='bilinear', scale_factor=(2, 1))
blocks = []
tmp_block = ProgressiveGeneratorBlock(
nn.Sequential(
weight_scale(nn.Linear(z_vars, base * (n_featuremaps)),
gain=calculate_gain('leaky_relu')), nn.LeakyReLU(0.2),
Reshape([[0], n_featuremaps, -1]),
create_conv_sequence(n_featuremaps, n_featuremaps)),
create_out_sequence(n_chans, n_featuremaps), create_fade_sequence(2))
blocks.append(tmp_block)
tmp_block = ProgressiveGeneratorBlock(
create_conv_sequence(n_featuremaps, n_featuremaps),
create_out_sequence(n_chans, n_featuremaps), create_fade_sequence(2))
blocks.append(tmp_block)
tmp_block = ProgressiveGeneratorBlock(
create_conv_sequence(n_featuremaps, n_featuremaps),
create_out_sequence(n_chans, n_featuremaps), None)
blocks.append(tmp_block)
return blocks
def create_out_sequence(n_chans, in_filters):
return nn.Sequential(
weight_scale(nn.Conv1d(in_filters, n_chans, 1),
gain=calculate_gain('linear')),
Reshape([[0], [1], [2], 1]), PixelShuffle2d([1, n_chans]))
def create_gen_blocks(n_chans,z_vars):
def create_conv_sequence(in_filters,out_filters):
conv_configs = list()
conv_configs.append({'kernel_size':3,'padding':1})
conv_configs.append({'kernel_size':5,'padding':2})
conv_configs.append({'kernel_size':7,'padding':3})
conv_configs.append({'kernel_size':9,'padding':4})
conv_configs.append({'kernel_size':11,'padding':5})
return nn.Sequential(nn.Upsample(mode='linear',scale_factor=2),
weight_scale(MultiConv1d(conv_configs,in_filters,out_filters,split_in_channels=True),
gain=calculate_gain('leaky_relu')),
nn.LeakyReLU(0.2),
PixelNorm(),
weight_scale(MultiConv1d(conv_configs,out_filters,out_filters,split_in_channels=True),
gain=calculate_gain('leaky_relu')),
nn.LeakyReLU(0.2),
PixelNorm())
def create_out_sequence(n_chans,in_filters):
return nn.Sequential(weight_scale(nn.Conv1d(in_filters,n_chans,1),
gain=calculate_gain('linear')),
Reshape([[0],[1],[2],1]),
PixelShuffle2d([1,n_chans]))
def create_fade_sequence(factor):
return nn.Upsample(mode='bilinear',scale_factor=(2,1))
blocks = []
tmp_block = ProgressiveGeneratorBlock(
nn.Sequential(weight_scale(nn.Linear(z_vars,50*12),
gain=calculate_gain('leaky_relu')),
nn.LeakyReLU(0.2),
Reshape([[0],50,-1]),
create_conv_sequence(50,50)),
create_out_sequence(n_chans,50),
create_fade_sequence(2)
)
blocks.append(tmp_block)
tmp_block = ProgressiveGeneratorBlock(
create_conv_sequence(50,50),
create_out_sequence(n_chans,50),
create_fade_sequence(2)
)
blocks.append(tmp_block)
tmp_block = ProgressiveGeneratorBlock(
create_conv_sequence(50,50),
create_out_sequence(n_chans,50),
create_fade_sequence(2)
)
blocks.append(tmp_block)
tmp_block = ProgressiveGeneratorBlock(
create_conv_sequence(50,50),
create_out_sequence(n_chans,50),
create_fade_sequence(2)
)
blocks.append(tmp_block)
tmp_block = ProgressiveGeneratorBlock(
create_conv_sequence(50,50),
create_out_sequence(n_chans,50),
create_fade_sequence(2)
)
blocks.append(tmp_block)
tmp_block = ProgressiveGeneratorBlock(
create_conv_sequence(50,50),
create_out_sequence(n_chans,50),
None
)
blocks.append(tmp_block)
return blocks
def create_disc_blocks(n_chans):
def create_conv_sequence(in_filters,out_filters,stdmap=False):
conv_configs = list()
conv_configs.append({'kernel_size':3,'padding':1})
conv_configs.append({'kernel_size':5,'padding':2})
conv_configs.append({'kernel_size':7,'padding':3})
conv_configs.append({'kernel_size':9,'padding':4})
conv_configs.append({'kernel_size':11,'padding':5})
filters_tmp = in_filters
split = True
if stdmap:
filters_tmp = in_filters-1
split = False
return nn.Sequential(weight_scale(MultiConv1d(conv_configs,in_filters,filters_tmp,split_in_channels=split),
gain=calculate_gain('leaky_relu')),
nn.LeakyReLU(0.2),
weight_scale(MultiConv1d(conv_configs,filters_tmp,out_filters,split_in_channels=True),
gain=calculate_gain('leaky_relu')),
nn.LeakyReLU(0.2),
weight_scale(nn.Conv1d(out_filters,out_filters,2,stride=2,groups=5),
gain=calculate_gain('leaky_relu')),
nn.LeakyReLU(0.2))
def create_in_sequence(n_chans,out_filters):
return nn.Sequential(weight_scale(nn.Conv2d(1,out_filters,(1,n_chans)),
gain=calculate_gain('leaky_relu')),
Reshape([[0],[1],[2]]),
nn.LeakyReLU(0.2))
def create_fade_sequence(factor):
return nn.AvgPool2d((factor,1),stride=(factor,1))
blocks = []
tmp_block = ProgressiveDiscriminatorBlock(
create_conv_sequence(50,50),
create_in_sequence(n_chans,50),
create_fade_sequence(2)
)
blocks.append(tmp_block)
tmp_block = ProgressiveDiscriminatorBlock(
create_conv_sequence(50,50),
create_in_sequence(n_chans,50),
create_fade_sequence(2)
)
blocks.append(tmp_block)
tmp_block = ProgressiveDiscriminatorBlock(
create_conv_sequence(50,50),
create_in_sequence(n_chans,50),
create_fade_sequence(2)
)
blocks.append(tmp_block)
tmp_block = ProgressiveDiscriminatorBlock(
create_conv_sequence(50,50),
create_in_sequence(n_chans,50),
create_fade_sequence(2)
)
blocks.append(tmp_block)
tmp_block = ProgressiveDiscriminatorBlock(
create_conv_sequence(50,50),
create_in_sequence(n_chans,50),
create_fade_sequence(2)
)
blocks.append(tmp_block)
tmp_block = ProgressiveDiscriminatorBlock(
nn.Sequential(StdMap1d(),
create_conv_sequence(51,50,stdmap=True),
Reshape([[0],-1]),
weight_scale(nn.Linear(50*12,1),
gain=calculate_gain('linear'))),
create_in_sequence(n_chans,50),
None
)
blocks.append(tmp_block)
return blocks