def __call__(self, x, is_training=False):
x = self.c1( x, is_training )
x = self.c2( x, is_training )
x = self.c3( x, is_training )
x1 = self.c4( x, is_training )
x2 = maxpool2d( x, 2, 2, padding="VALID" )
x = concat( [ x1, x2 ], axis = 3, name = "{}_concat1".format( self.name ) )
x1 = self.c5_0( x, is_training )
x2 = self.c6_0( x, is_training )
x1 = self.c5_1( x1, is_training )
x1 = self.c5_2( x1, is_training )
x1 = self.c5_3( x1, is_training )
x2 = self.c6_1( x2, is_training )
x = concat( [ x1, x2 ], axis = 3, name = "{}_concat2".format( self.name ) )
x1 = self.c7( x, is_training )
x2 = maxpool2d( x, 2, 2, padding="VALID" )
x = concat( [ x1, x2 ], axis = 3, name = "{}_concat3".format( self.name ) )
print(x)
return x
def __call__(self, x, is_training=False):
x0 = self.c0_0( x, is_training )
x1 = self.c1_0( x, is_training )
x2 = self.c2_0( x, is_training )
x1 = self.c1_1( x1, is_training )
x2 = self.c2_1( x2, is_training )
x2 = self.c2_2( x2, is_training )
xf = concat( [ x0, x1, x2 ], 3, name = "{}_concat".format( self.name ) )
xf = self.c3_0( xf, is_training )
xf = self.se( xf, is_training )
x = x + xf
if self.act:
x = self.act( x )
print(x)
return x
def __call__(self, x, is_training=False):
concat_feat = x
for i, l in enumerate(self.layers):
x = l(concat_feat, is_training)
concat_feat = concat([concat_feat, x],
axis=3,
name="{}_{}_concat".format(self.name, i))
print(concat_feat)
return concat_feat
def __call__(self, x, is_training=False):
x0 = maxpool2d( x, 3, 2, padding="SAME" )
x1 = self.c0_0( x, is_training )
x2 = self.c1_0( x, is_training )
x2 = self.c1_1( x2, is_training )
x2 = self.c1_2( x2, is_training )
x = concat( [ x0, x1, x2 ], 3, name = "{}_concat".format( self.name ) )
print(x)
return x
