def graph_fn(image, boxes, levels): return spatial_ops.multilevel_roi_align([image], boxes, levels, output_size=[1, 1], num_samples_per_cell_y=2, num_samples_per_cell_x=2)
def graph_fn(feature1, feature2, feature3, feature4, boxes, levels): roi_features = spatial_ops.multilevel_roi_align( [feature1, feature2, feature3, feature4], boxes, levels, output_size) return roi_features
def crop_and_resize_fn(): tf_features = [ tf.constant(feature, dtype=tf.bfloat16) for feature in features ] return spatial_ops.multilevel_roi_align( tf_features, tf.constant(boxes), tf.constant(levels), output_size)
def crop_and_resize_fn(): return spatial_ops.multilevel_roi_align( features, tf_boxes, tf_levels, output_size)
def graph_fn(image, boxes, levels): return spatial_ops.multilevel_roi_align([image], boxes, levels, output_size=[3, 3], extrapolation_value=2.0)
def graph_fn(image, boxes, levels): return spatial_ops.multilevel_roi_align([image], boxes, levels, output_size=[3, 3])