def __init__(self, in_ch=3, n_down_layers=4):
layers = {}
w = chainer.initializers.Normal(0.02)
self.n_down_layers = n_down_layers
layers['c0'] = CBR(in_ch, 64, bn=False, sample='down', activation=F.leaky_relu, dropout=False, noise=True)
base = 64
for i in range(1, n_down_layers):
layers['c'+str(i)] = CBR(base, base*2, bn=True, sample='down', activation=F.leaky_relu, dropout=False, noise=True)
base*=2
layers['c'+str(n_down_layers)] = CBR(base, 1, bn=False, sample='none', activation=None, dropout=False, noise=True)
super(Discriminator, self).__init__(**layers)
评论列表
文章目录