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 | Четырехмерный float Tensor формы [num_images, height, width, num_achors], содержащий оценки прямоугольников для заданных якорей, может быть неотсортированным. |
bbox_deltas | Четырехмерный float Tensor формы [num_images, height, width, 4 x num_anchors] , кодирующий прямоугольники относительно каждого якоря. Координаты задаются в форме [dy, dx, dh, dw]. |
image_info | Двухмерный float Tensor формы [num_images, 5] , содержащий информацию об изображении Высота, Ширина, Масштаб. |
anchors | Двухмерный 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 | Оценки прямоугольников ROI в их Tensor. |
© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.4/api_docs/python/tf/image/generate_bounding_box_proposals