@Operator public final class NonMaxSuppressionV3 extends PrimitiveOp implements Operand<Integer>
pruning away boxes that have high intersection-over-union (IOU) overlap with previously selected boxes. Bounding boxes with score less than `score_threshold` are removed. Bounding boxes are supplied as [y1, x1, y2, x2], where (y1, x1) and (y2, x2) are the coordinates of any diagonal pair of box corners and the coordinates can be provided as normalized (i.e., lying in the interval [0, 1]) or absolute. Note that this algorithm is agnostic to where the origin is in the coordinate system and more generally is invariant to orthogonal transformations and translations of the coordinate system; thus translating or reflections of the coordinate system result in the same boxes being selected by the algorithm. The output of this operation is a set of integers indexing into the input collection of bounding boxes representing the selected boxes. The bounding box coordinates corresponding to the selected indices can then be obtained using the `tf.gather operation`. For example: selected_indices = tf.image.non_max_suppression_v2( boxes, scores, max_output_size, iou_threshold, score_threshold) selected_boxes = tf.gather(boxes, selected_indices)
operation| Modifier and Type | Method and Description |
|---|---|
Output<Integer> |
asOutput()
Returns the symbolic handle of a tensor.
|
static <T extends Number> |
create(Scope scope,
Operand<T> boxes,
Operand<T> scores,
Operand<Integer> maxOutputSize,
Operand<Float> iouThreshold,
Operand<Float> scoreThreshold)
Factory method to create a class to wrap a new NonMaxSuppressionV3 operation to the graph.
|
Output<Integer> |
selectedIndices()
A 1-D integer tensor of shape `[M]` representing the selected
indices from the boxes tensor, where `M <= max_output_size`.
|
equals, hashCode, toStringpublic static <T extends Number> NonMaxSuppressionV3 create(Scope scope, Operand<T> boxes, Operand<T> scores, Operand<Integer> maxOutputSize, Operand<Float> iouThreshold, Operand<Float> scoreThreshold)
scope - current graph scopeboxes - A 2-D float tensor of shape `[num_boxes, 4]`.scores - A 1-D float tensor of shape `[num_boxes]` representing a single
score corresponding to each box (each row of boxes).maxOutputSize - A scalar integer tensor representing the maximum number of
boxes to be selected by non max suppression.iouThreshold - A 0-D float tensor representing the threshold for deciding whether
boxes overlap too much with respect to IOU.scoreThreshold - A 0-D float tensor representing the threshold for deciding when to remove
boxes based on score.public Output<Integer> selectedIndices()
public Output<Integer> asOutput()
OperandInputs to TensorFlow operations are outputs of another TensorFlow operation. This method is used to obtain a symbolic handle that represents the computation of the input.
asOutput in interface Operand<Integer>OperationBuilder.addInput(Output)Copyright © 2015–2019. All rights reserved.