tfm.vision.box_ops.filter_boxes
Stay organized with collections
Save and categorize content based on your preferences.
Filters and remove boxes that are too small or fall outside the image.
tfm.vision.box_ops.filter_boxes(
boxes, scores, image_shape, min_size_threshold
)
Args |
boxes
|
a tensor whose last dimension is 4 representing the coordinates of
boxes in ymin, xmin, ymax, xmax order.
|
scores
|
a tensor whose shape is the same as tf.shape(boxes)[:-1]
representing the original scores of the boxes.
|
image_shape
|
a tensor whose shape is the same as, or broadcastable to
boxes except the last dimension, which is 2, representing [height,
width] of the scaled image.
|
min_size_threshold
|
a float representing the minimal box size in each side
(w.r.t. the scaled image). Boxes whose sides are smaller than it will be
filtered out.
|
Returns |
filtered_boxes
|
a tensor whose shape is the same as boxes but with
the position of the filtered boxes are filled with 0.
|
filtered_scores
|
a tensor whose shape is the same as 'scores' but with
the positinon of the filtered boxes filled with 0.
|
Except as otherwise noted, the content of this page is licensed under the Creative Commons Attribution 4.0 License, and code samples are licensed under the Apache 2.0 License. For details, see the Google Developers Site Policies. Java is a registered trademark of Oracle and/or its affiliates. Some content is licensed under the numpy license.
Last updated 2024-02-02 UTC.
[[["Easy to understand","easyToUnderstand","thumb-up"],["Solved my problem","solvedMyProblem","thumb-up"],["Other","otherUp","thumb-up"]],[["Missing the information I need","missingTheInformationINeed","thumb-down"],["Too complicated / too many steps","tooComplicatedTooManySteps","thumb-down"],["Out of date","outOfDate","thumb-down"],["Samples / code issue","samplesCodeIssue","thumb-down"],["Other","otherDown","thumb-down"]],["Last updated 2024-02-02 UTC."],[],[]]