Python FasterRCNNConfig Beispiele

Beispiel #1

Datei: faster_rcnn_res50_coco_1x_800size.py Projekt: 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)

Beispiel #2

Datei: faster_rcnn_res50_coco_1x_800size.py Projekt: 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)