Exemplo n.º 1
0
 def generate_constrained_output_sub_blocks(self, input_shape):
     return [
         TwoDClassificationBlock(
             input_shape=input_shape,
             parent_block=self,
             class_count=self.class_count,
         )
     ]
Exemplo n.º 2
0
 def generate_constrained_output_sub_blocks(self, input_shape):
     return [
         TwoDClassificationBlock(
             input_shape=input_shape,
             parent_block=self,
             class_count=self.class_count,
             layer_type=self.SUB_BLOCK_TYPES.CLASSIFICATION_BLOCK,
         )
     ]