Commit d8611151 authored by syiming's avatar syiming
Browse files

change _crop_and_resize_fn to multilevel ones

parent 8a120d0e
...@@ -35,6 +35,7 @@ from object_detection.protos import box_predictor_pb2 ...@@ -35,6 +35,7 @@ from object_detection.protos import box_predictor_pb2
from object_detection.protos import hyperparams_pb2 from object_detection.protos import hyperparams_pb2
from object_detection.protos import post_processing_pb2 from object_detection.protos import post_processing_pb2
from object_detection.utils import ops from object_detection.utils import ops
from object_detection.utils import spatial_transform_ops as spatial_ops
from object_detection.utils import test_case from object_detection.utils import test_case
from object_detection.utils import test_utils from object_detection.utils import test_utils
from object_detection.utils import tf_version from object_detection.utils import tf_version
...@@ -377,8 +378,9 @@ class FasterRCNNMetaArchTestBase(test_case.TestCase, parameterized.TestCase): ...@@ -377,8 +378,9 @@ class FasterRCNNMetaArchTestBase(test_case.TestCase, parameterized.TestCase):
max_negatives_per_positive=None) max_negatives_per_positive=None)
crop_and_resize_fn = ( crop_and_resize_fn = (
ops.matmul_crop_and_resize spatial_ops.multilevel_matmul_crop_and_resize
if use_matmul_crop_and_resize else ops.native_crop_and_resize) if use_matmul_crop_and_resize
else spatial_ops.multilevel_native_crop_and_resize)
common_kwargs = { common_kwargs = {
'is_training': 'is_training':
is_training, is_training,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment