def __init__(self, channel=128):
super(exp_4, self).__init__()
self.channel = channel
self.encoder = encoder()
self.decoder_0 = self._make_layer(Bottleneck,
channel // 4,
3,
stride=1)
self.decoder_1 = self._make_layer(Bottleneck,
channel // 4,
3,
stride=1)
self.decoder_2 = self._make_layer(Bottleneck,
channel // 4,
3,
stride=1)
self.decoder_3 = self._make_layer(Bottleneck,
channel // 4,
3,
stride=1)
self.decoder_4 = self._make_layer(Bottleneck,
channel // 4,
3,
stride=1)
self.decoder_5 = self._make_layer(Bottleneck,
channel // 4,
3,
stride=1)
self.ppm = PPM(channel)
self.output_s = ConcatOutput(channel)
weight_init(self)
def __init__(self, channel=128):
super(exp_1, self).__init__()
self.encoder = encoder()
self.reduce_s0 = Reduction(128, channel)
self.reduce_s1 = Reduction(256, channel)
self.reduce_s2 = Reduction(512, channel)
self.reduce_s3 = Reduction(1024, channel)
self.reduce_s4 = Reduction(2048, channel)
self.ppm = PPM(channel)
self.output_s = ConcatOutput(channel)
weight_init(self)
def __init__(self, channel=128):
super(exp_2, self).__init__()
self.channel = channel
self.encoder = encoder()
self.reduce_s0 = Reduction(128, channel)
self.reduce_s1 = Reduction(256, channel)
self.reduce_s2 = Reduction(512, channel)
self.reduce_s3 = Reduction(1024, channel)
self.reduce_s4 = Reduction(2048, channel)
self.decoder_0 = self._make_layer(Bottleneck, channel // 4, 3, stride=1)
self.decoder_1 = self._make_layer(Bottleneck, channel // 4, 3, stride=1)
self.decoder_2 = self._make_layer(Bottleneck, channel // 4, 3, stride=1)
self.decoder_3 = self._make_layer(Bottleneck, channel // 4, 3, stride=1)
self.decoder_4 = self._make_layer(Bottleneck, channel // 4, 3, stride=1)
self.ppm = PPM(channel)
self.output_s = ConcatOutput(channel)
weight_init(self)