Spaces:
Running
Running
| ##+++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++ | |
| ## Created by: Donny You, RainbowSecret | |
| ## Microsoft Research | |
| ## [email protected] | |
| ## Copyright (c) 2019 | |
| ## | |
| ## This source code is licensed under the MIT-style license found in the | |
| ## LICENSE file in the root directory of this source tree | |
| ##+++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++ | |
| from __future__ import absolute_import | |
| from __future__ import division | |
| from __future__ import print_function | |
| # from lib.models.backbones.resnet.resnet_backbone import ResNetBackbone | |
| # from lib.models.backbones.hrnet.hrnet_backbone import HRNetBackbone | |
| from .hrt.hrt_backbone import HRTBackbone | |
| # from lib.models.backbones.swin.swin_backbone import SwinTransformerBackbone | |
| from .hrt.logger import Logger as Log | |
| class BackboneSelector(object): | |
| def __init__(self, configer): | |
| self.configer = configer | |
| def get_backbone(self, **params): | |
| backbone = self.configer.get("network", "backbone") | |
| model = None | |
| # if ( | |
| # "resnet" in backbone or "resnext" in backbone or "resnest" in backbone | |
| # ) and "senet" not in backbone: | |
| # model = ResNetBackbone(self.configer)(**params) | |
| if "hrt" in backbone: | |
| # model = HRTBackbone(self.configer)(**params) | |
| pass | |
| # elif "hrnet" in backbone: | |
| # model = HRNetBackbone(self.configer)(**params) | |
| # elif "swin" in backbone: | |
| # model = SwinTransformerBackbone(self.configer)(**params) | |
| else: | |
| Log.error("Backbone {} is invalid.".format(backbone)) | |
| exit(1) | |
| return model | |
| class Test(): | |
| def __init__(): | |
| pass |