Crop masks in target boxes.
tfm.vision.spatial_transform_ops.crop_mask_in_target_box(
masks, boxes, target_boxes, output_size, sample_offset=0, use_einsum=True
)
Args |
masks
|
A tensor with a shape of [batch_size, num_masks, height, width].
|
boxes
|
a float tensor representing box cooridnates that tightly enclose
masks with a shape of [batch_size, num_masks, 4] in un-normalized
coordinates. A box is represented by [ymin, xmin, ymax, xmax].
|
target_boxes
|
a float tensor representing target box cooridnates for masks
with a shape of [batch_size, num_masks, 4] in un-normalized coordinates. A
box is represented by [ymin, xmin, ymax, xmax].
|
output_size
|
A scalar to indicate the output crop size. It currently only
supports to output a square shape outputs.
|
sample_offset
|
a float number in [0, 1] indicates the subpixel sample offset
from grid point.
|
use_einsum
|
Use einsum to replace gather in selective_crop_and_resize.
|
Returns |
A 4-D tensor representing feature crop of shape
[batch_size, num_boxes, output_size, output_size].
|