Skip to content

GitLab

Commit 677aaa11 authored by Austin Myers's avatar Austin Myers Committed by TF Object Detection Team
Browse files

Enables combined static and dynamic shape for the batch size in 'MultiLevelRoIAlign' 

PiperOrigin-RevId: 377103499
parent 213a9649
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 1 deletion
research/object_detection/utils/spatial_transform_ops.py
View file @ 677aaa11
...@@ -19,6 +19,7 @@ from __future__ import division ...@@ -19,6 +19,7 @@ from __future__ import division
from __future__ import print_function from __future__ import print_function
import tensorflow.compat.v1 as tf import tensorflow.compat.v1 as tf
from object_detection.utils import shape_utils
def _coordinate_vector_1d(start, end, size, align_endpoints): def _coordinate_vector_1d(start, end, size, align_endpoints):
...@@ -322,7 +323,7 @@ def multilevel_roi_align(features, boxes, box_levels, output_size, ...@@ -322,7 +323,7 @@ def multilevel_roi_align(features, boxes, box_levels, output_size,
""" """
with tf.name_scope(scope, 'MultiLevelRoIAlign'): with tf.name_scope(scope, 'MultiLevelRoIAlign'):
features, true_feature_shapes = pad_to_max_size(features) features, true_feature_shapes = pad_to_max_size(features)
batch_size = tf.shape(features)[0] batch_size = shape_utils.combined_static_and_dynamic_shape(features)[0]
num_levels = features.get_shape().as_list()[1] num_levels = features.get_shape().as_list()[1]
max_feature_height = tf.shape(features)[2] max_feature_height = tf.shape(features)[2]
max_feature_width = tf.shape(features)[3] max_feature_width = tf.shape(features)[3]
......
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