def generate_constrained_output_sub_blocks(self, input_shape):
return [
TwoDClassificationBlock(
input_shape=input_shape,
parent_block=self,
class_count=self.class_count,
)
]
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,
)
]