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