| # -*- coding: utf-8 -*- | |
| # @Time : 2019/8/23 21:54 | |
| # @Author : zhoujun | |
| from .resnet import * | |
| __all__ = ["build_backbone"] | |
| support_backbone = [ | |
| "resnet18", | |
| "deformable_resnet18", | |
| "deformable_resnet50", | |
| "resnet50", | |
| "resnet34", | |
| "resnet101", | |
| "resnet152", | |
| ] | |
| def build_backbone(backbone_name, **kwargs): | |
| assert ( | |
| backbone_name in support_backbone | |
| ), f"all support backbone is {support_backbone}" | |
| backbone = eval(backbone_name)(**kwargs) | |
| return backbone | |