Spaces:
Runtime error
Runtime error
File size: 642 Bytes
c709b60 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 |
# Copyright (c) Facebook, Inc. and its affiliates.
import unittest
from detectron2.config import get_cfg
from detectron2.export.c10 import Caffe2RPN
from detectron2.layers import ShapeSpec
class TestCaffe2RPN(unittest.TestCase):
def test_instantiation(self):
cfg = get_cfg()
cfg.MODEL.RPN.BBOX_REG_WEIGHTS = (1, 1, 1, 1, 1)
input_shapes = {'res4': ShapeSpec(channels=256, stride=4)}
rpn = Caffe2RPN(cfg, input_shapes)
assert rpn is not None
cfg.MODEL.RPN.BBOX_REG_WEIGHTS = (10, 10, 5, 5, 1)
with self.assertRaises(AssertionError):
rpn = Caffe2RPN(cfg, input_shapes)
|