test_FasterRCNN.py 文件源码

def test_forward_cpu_VGG16(self):
        print('test_forward_cpu_VGG16')
        gpu = -1
        trunk = VGG16
        rpn_in_ch = 512
        rpn_out_ch = 512
        n_anchors = 9
        feat_stride = 16
        anchor_scales = [8, 16, 32]
        num_classes = 21
        spatial_scale = 0.0625
        model = FasterRCNN(
            gpu, trunk, rpn_in_ch, rpn_out_ch, n_anchors, feat_stride,
            anchor_scales, num_classes, spatial_scale)

        model.train = False
        ret = model(chainer.Variable(self.x, volatile=True), self.im_info)
        assert(len(ret) == 2)
        assert(isinstance(ret[0], chainer.Variable))
        assert(isinstance(ret[1], np.ndarray))