예제 #1
0
    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
예제 #2
0
    def resnet_c4c5_factory(cls, depth, use_3x3_conv0, use_bn_preprocess,
                            num_branch, branch_dilates, branch_ids, branch_bn_shared, branch_conv_shared, branch_deform,
                            norm_type="local", norm_mom=0.9, ndev=None, fp16=False):
        c1, c2, c3, c4, c5 = cls.resnet_factory(depth, use_3x3_conv0, use_bn_preprocess,
                                                num_branch, branch_dilates, branch_ids, branch_bn_shared, branch_conv_shared, branch_deform,
                                                norm_type, norm_mom, ndev, fp16)
        c5 = X.fixbn(c5, "bn1")
        c5 = X.relu(c5)

        return c4, c5