コード例 #1
0
ファイル: factory.py プロジェクト: zxzxzxygithub/models-1
def mask_rcnn_head_generator(params):
    """Generator function for Mask R-CNN head architecture."""
    return heads.MaskrcnnHead(params.num_classes,
                              params.mask_target_size,
                              params.num_convs,
                              params.num_filters,
                              params.use_separable_conv,
                              params.use_batch_norm,
                              batch_norm_relu=batch_norm_relu_generator(
                                  params.batch_norm))
コード例 #2
0
ファイル: factory.py プロジェクト: tpsgrp/python-app
def mask_rcnn_head_generator(params):
    """Generator function for Mask R-CNN head architecture."""
    head_params = params.mrcnn_head
    return heads.MaskrcnnHead(params.architecture.num_classes,
                              params.architecture.mask_target_size,
                              head_params.num_convs,
                              head_params.num_filters,
                              head_params.use_separable_conv,
                              params.norm_activation.activation,
                              head_params.use_batch_norm,
                              norm_activation=norm_activation_generator(
                                  params.norm_activation))
コード例 #3
0
def mask_rcnn_head_generator(params):
    """Generator function for Mask R-CNN head architecture."""
    return heads.MaskrcnnHead(params.num_classes,
                              params.mrcnn_resolution,
                              batch_norm_relu=batch_norm_relu_generator(
                                  params.batch_norm))