Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
ModelZoo
ResNet50_tensorflow
Commits
d8611151
Commit
d8611151
authored
Jul 01, 2020
by
syiming
Browse files
change _crop_and_resize_fn to multilevel ones
parent
8a120d0e
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
4 additions
and
2 deletions
+4
-2
research/object_detection/meta_architectures/faster_rcnn_meta_arch_test_lib.py
...tion/meta_architectures/faster_rcnn_meta_arch_test_lib.py
+4
-2
No files found.
research/object_detection/meta_architectures/faster_rcnn_meta_arch_test_lib.py
View file @
d8611151
...
...
@@ -35,6 +35,7 @@ from object_detection.protos import box_predictor_pb2
from
object_detection.protos
import
hyperparams_pb2
from
object_detection.protos
import
post_processing_pb2
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_utils
from
object_detection.utils
import
tf_version
...
...
@@ -377,8 +378,9 @@ class FasterRCNNMetaArchTestBase(test_case.TestCase, parameterized.TestCase):
max_negatives_per_positive
=
None
)
crop_and_resize_fn
=
(
ops
.
matmul_crop_and_resize
if
use_matmul_crop_and_resize
else
ops
.
native_crop_and_resize
)
spatial_ops
.
multilevel_matmul_crop_and_resize
if
use_matmul_crop_and_resize
else
spatial_ops
.
multilevel_native_crop_and_resize
)
common_kwargs
=
{
'is_training'
:
is_training
,
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment