Python FasterRCNNの例

コード例 #1

ファイル: faster_rcnn_res50_coco_1x_800size.py プロジェクト: bigeagle/Models

def faster_rcnn_res50_coco_1x_800size(batch_size=1, **kwargs):
    r"""
    Faster-RCNN FPN trained from COCO dataset.
    `"Faster-RCNN" <https://arxiv.org/abs/1506.01497>`_
    `"FPN" <https://arxiv.org/abs/1612.03144>`_
    `"COCO" <https://arxiv.org/abs/1405.0312>`_
    """
    return models.FasterRCNN(models.FasterRCNNConfig(), batch_size=batch_size, **kwargs)

コード例 #2

ファイル: faster_rcnn_res50_objects365_1x_800size.py プロジェクト: klsdjft/Models-1

def faster_rcnn_res50_objects365_1x_800size(batch_size=1, **kwargs):
    r"""
    Faster-RCNN FPN trained from Objects365 dataset.
    `"Faster-RCNN" <https://arxiv.org/abs/1506.01497>`_
    `"FPN" <https://arxiv.org/abs/1612.03144>`_
    """
    cfg = CustomFasterRCNNConfig()
    cfg.backbone_pretrained = False
    return models.FasterRCNN(cfg, batch_size=batch_size, **kwargs)

コード例 #3

ファイル: faster_rcnn_res50_coco_1x_800size.py プロジェクト: zihan987/Models

def faster_rcnn_res50_coco_1x_800size(**kwargs):
    r"""
    Faster-RCNN FPN trained from COCO dataset.
    `"Faster-RCNN" <https://arxiv.org/abs/1506.01497>`_
    `"FPN" <https://arxiv.org/abs/1612.03144>`_
    `"COCO" <https://arxiv.org/abs/1405.0312>`_
    """
    cfg = models.FasterRCNNConfig()
    cfg.backbone_pretrained = False
    return models.FasterRCNN(cfg, **kwargs)

コード例 #4

ファイル: faster_rcnn_res50_coco_1x_800size_syncbn.py プロジェクト: klsdjft/Models-1

def faster_rcnn_res50_coco_1x_800size_syncbn(batch_size=1, **kwargs):
    r"""
    Faster-RCNN FPN trained from COCO dataset.
    `"Faster-RCNN" <https://arxiv.org/abs/1506.01497>`_
    `"FPN" <https://arxiv.org/abs/1612.03144>`_
    `"COCO" <https://arxiv.org/abs/1405.0312>`_
    `"SyncBN" <https://arxiv.org/abs/1711.07240>`_
    """
    cfg = CustomFasterRCNNConfig()
    cfg.backbone_pretrained = False
    return models.FasterRCNN(cfg, batch_size=batch_size, **kwargs)