tf.image.generate_bounding_box_proposals
Создает предложения ограничивающих рамок из закодированных ограничивающих рамок.
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 | Скалярный int Tensor, задающий количество рамок с наивысшими оценками, используемых в качестве входных данных. По умолчанию — 6000. |
min_size | Скалярный float Tensor. Любая рамка, размер которой меньше min_size, будет отброшена. По умолчанию — 16. |
post_nms_topn | Целое число. Максимальное количество областей интереса в выходных данных. |
name | Имя этой операции (необязательно). |
| Возвращаемые значения | |
|---|---|
rois | Рамки областей интереса, отсортированные по оценкам. |
roi_probabilities | Оценки рамок областей интереса в порядке 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/api_docs/python/tf/image/generate_bounding_box_proposals