|
|
|
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
|
|
|
|
|
|
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)
|
|
|
