test_anchor_target_layer.py 文件源码

def test_call(self):
        xp = chainer.cuda.cupy
        x = chainer.Variable(xp.asarray(self.x, dtype=xp.float32))
        gt_boxes = self.gt_boxes
        im_info = self.im_info
        labels, bbox_targets, bbox_inside_weights, bbox_outside_weights = \
            self.anchor_target_layer(x, gt_boxes, im_info)

        n_anchors = self.anchor_target_layer.n_anchors
        self.assertEqual(labels.shape,
                         (1, n_anchors, self.height, self.width))
        self.assertEqual(bbox_targets.shape,
                         (1, n_anchors * 4, self.height, self.width))
        self.assertEqual(bbox_inside_weights.shape,
                         (1, n_anchors * 4, self.height, self.width))
        self.assertEqual(bbox_outside_weights.shape,
                         (1, n_anchors * 4, self.height, self.width))