remove assert for AnchorsGenerator

wz · wz · commit df8f9b7a7049 · 2022-04-26T12:24:25.000+08:00
diff --git a/pytorch_object_detection/mask_rcnn/network_files/faster_rcnn_framework.py b/pytorch_object_detection/mask_rcnn/network_files/faster_rcnn_framework.py
@@ -269,7 +269,7 @@ def __init__(self, backbone, num_classes=None,
                 "same for all the levels"
             )
 
-        assert isinstance(rpn_anchor_generator, (AnchorsGenerator, type(None)))
+        # assert isinstance(rpn_anchor_generator, (AnchorsGenerator, type(None)))
         assert isinstance(box_roi_pool, (MultiScaleRoIAlign, type(None)))
 
         if num_classes is not None:

Original file line number	Diff line number	Diff line change
`@@ -269,7 +269,7 @@ def __init__(self, backbone, num_classes=None,`
`269`	`269`	`"same for all the levels"`
`270`	`270`	`)`
`271`	`271`
`272`		`- assert isinstance(rpn_anchor_generator, (AnchorsGenerator, type(None)))`
	`272`	`+ # assert isinstance(rpn_anchor_generator, (AnchorsGenerator, type(None)))`
`273`	`273`	`assert isinstance(box_roi_pool, (MultiScaleRoIAlign, type(None)))`
`274`	`274`
`275`	`275`	`if num_classes is not None:`