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)