def __init__(self, pBackbone): super().__init__(pBackbone) from mxnext.backbone.resnext import Builder b = Builder() self.c4, self.c5 = b.get_backbone("mxnet", 101, "c4c5", pBackbone.normalizer, pBackbone.num_group, pBackbone.fp16)
def _get_bbox_head_logit(self, conv_feat): if self._head_feat is not None: return self._head_feat from mxnext.backbone.resnet_v2 import Builder unit = Builder.resnet_stage( conv_feat, name="stage4", num_block=3, filter=2048, stride=1, dilate=1, norm_type=self.p.normalizer, norm_mom=0.9, ndev=8 ) bn1 = X.fixbn(unit, name='bn1') relu1 = X.relu(bn1, name='relu1') relu1 = X.to_fp32(relu1, name='c5_to_fp32') pool1 = X.pool(relu1, global_pool=True, name='pool1') self._head_feat = pool1 return self._head_feat
def _get_bbox_head_logit(self, conv_feat): if self._head_feat is not None: return self._head_feat from mxnext.backbone.resnext import Builder unit = Builder.resnext_stage(conv_feat, name="stage4", num_block=3, filter=2048, stride=1, dilate=1, num_group=self.p.num_group, norm_type=self.p.normalizer, norm_mom=0.9, ndev=8) pool1 = X.pool(unit, global_pool=True, name='pool1') self._head_feat = pool1 return self._head_feat
def __init__(self, pBackbone): super().__init__(pBackbone) from mxnext.backbone.resnet_v1d import Builder b = Builder() self.symbol = b.get_backbone(self.p.depth, "c4", pBackbone.normalizer, pBackbone.fp16)
def __init__(self, pBackbone): super().__init__(pBackbone) from mxnext.backbone.resnext import Builder b = Builder() self.symbol = b.get_backbone("mxnet", 50, "c4", pBackbone.normalizer, pBackbone.num_group, pBackbone.fp16)
def __init__(self, pBackbone): super(MXNetResNet101V2C4C5, self).__init__(pBackbone) from mxnext.backbone.resnet_v2 import Builder b = Builder() self.c4, self.c5 = b.get_backbone("mxnet", 101, "c4c5", pBackbone.normalizer, pBackbone.fp16)
def __init__(self, pBackbone): super(ResNet101V1, self).__init__(pBackbone) from mxnext.backbone.resnet_v1 import Builder b = Builder() self.symbol = b.get_backbone("msra", 101, "c4", pBackbone.normalizer, pBackbone.fp16)
def __init__(self, pBackbone): super(MXNetResNet50V2, self).__init__(pBackbone) from mxnext.backbone.resnet_v2 import Builder b = Builder() self.symbol = b.get_backbone("mxnet", 50, "c4", pBackbone.normalizer, pBackbone.fp16)