tf.image.generate_bounding_box_proposals
Генерация предложений bounding box из закодированных bounding box.
tf.image.generate_bounding_box_proposals(
scores,
bbox_deltas,
image_info,
anchors,
nms_threshold=0.7,
pre_nms_topn=6000,
min_size=16,
post_nms_topn=300,
name=None
)
| Аргументы | |
|---|---|
scores | 4-мерный float Tensor формы [num_images, height, width, num_achors], содержащий оценки ящиков для заданных якорей, может быть неотсортированным. |
bbox_deltas | 4-мерный float Tensor формы [num_images, height, width, 4 x num_anchors] кодирования ящиков относительно каждого якоря. Координаты заданы в форме [dy, dx, dh, dw]. |
image_info | 2-мерный float Tensor формы [num_images, 5] содержащий информацию о изображении Высота, Ширина, Масштаб. |
anchors | 2-мерный float Tensor формы [num_anchors, 4] описывающий ящики якорей. Ящики отформатированы в форме [y1, x1, y2, x2]. |
nms_threshold | Скалярный float Tensor для порога не-максимального подавления. По умолчанию 0,7. |
pre_nms_topn | Скалярное целое число Tensor для количества ящиков с наибольшим значением оценки, которые будут использоваться в качестве входных данных. По умолчанию 6000. |
min_size | Скалярный float Tensor. Любой ящик, имеющий меньший размер, чем min_size, будет отброшен. По умолчанию 16. |
post_nms_topn | Целое число. Максимальное количество ROI в выходных данных. |
name | Имя для этой операции (необязательно). |
| Возвращаемые значения | |
|---|---|
rois | Области интереса, отсортированные по их оценкам. |
roi_probabilities | Оценки ящиков ROI в их Tensor. |
© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.9/api_docs/python/tf/image/generate_bounding_box_proposals