Spaces:
Runtime error
Runtime error
| # Copyright (c) Facebook, Inc. and its affiliates. | |
| from detectron2.utils.registry import Registry | |
| PROPOSAL_GENERATOR_REGISTRY = Registry("PROPOSAL_GENERATOR") | |
| PROPOSAL_GENERATOR_REGISTRY.__doc__ = """ | |
| Registry for proposal generator, which produces object proposals from feature maps. | |
| The registered object will be called with `obj(cfg, input_shape)`. | |
| The call should return a `nn.Module` object. | |
| """ | |
| from . import rpn, rrpn # noqa F401 isort:skip | |
| def build_proposal_generator(cfg, input_shape): | |
| """ | |
| Build a proposal generator from `cfg.MODEL.PROPOSAL_GENERATOR.NAME`. | |
| The name can be "PrecomputedProposals" to use no proposal generator. | |
| """ | |
| name = cfg.MODEL.PROPOSAL_GENERATOR.NAME | |
| if name == "PrecomputedProposals": | |
| return None | |
| return PROPOSAL_GENERATOR_REGISTRY.get(name)(cfg, input_shape) | |