def retinanet_head_generator(params):
"""Generator function for RetinaNet head architecture."""
return heads.RetinanetHead(params.min_level,
params.max_level,
params.num_classes,
params.anchors_per_location,
params.retinanet_head_num_convs,
params.retinanet_head_num_filters,
batch_norm_relu=batch_norm_relu_generator(
params.batch_norm))
def retinanet_head_generator(params):
"""Generator function for RetinaNet head architecture."""
head_params = params.retinanet_head
return heads.RetinanetHead(params.architecture.min_level,
params.architecture.max_level,
params.architecture.num_classes,
head_params.anchors_per_location,
head_params.num_convs,
head_params.num_filters,
head_params.use_separable_conv,
norm_activation=norm_activation_generator(
params.norm_activation))