9 lines
198 B
Python
Executable File
9 lines
198 B
Python
Executable File
from ssd.modeling import registry
|
|
from .box_head import SSDBoxHead
|
|
|
|
__all__ = ['build_box_head', 'SSDBoxHead']
|
|
|
|
|
|
def build_box_head(cfg):
|
|
return registry.BOX_HEADS[cfg.MODEL.BOX_HEAD.NAME](cfg)
|