tf.contrib.image.bipartite_match
Найти двудольное соответствие на основе заданной матрицы расстояний.
tf.contrib.image.bipartite_match(
distance_mat, num_valid_rows, top_k=-1, name='bipartite_match'
)
Используется жадный алгоритм двудольного сопоставления для получения соответствия с (жадным) минимальным расстоянием.
| Аргументы | |
|---|---|
distance_mat | Матрица-тензор 2-D с плавающей запятой формы [num_rows, num_columns]. Это матрица парных расстояний между сущностями, представленными каждой строкой и каждым столбцом. Это асимметричная матрица. Чем меньше расстояние, тем более похожи пары. Двудольное сопоставление направлено на минимизацию расстояний. |
num_valid_rows | Скаляр или 1-мерный тензор с одним элементом, описывающим количество допустимых строк матрицы расстояний distance_mat, которые следует учитывать для двудольного сопоставления. Если значение установлено в отрицательное, то используются все строки из distance_mat. |
top_k | Скаляр, который указывает количество top-k совпадений для извлечения. Если значение установлено в отрицательное, то оно устанавливается в соответствии с максимальным числом совпадений из distance_mat. |
name | Имя операции. |
| Возвращаемые значения | |
|---|---|
row_to_col_match_indices | Вектор длиной num_rows, который представляет количество строк входной distance_matrix. Если row_to_col_match_indices[i] не -1, строка i сопоставляется со столбцом row_to_col_match_indices[i]. |
col_to_row_match_indices | Вектор длиной num_columns, который представляет количество столбцов входной матрицы расстояний. Если col_to_row_match_indices[j] не -1, столбец j сопоставляется со строкой col_to_row_match_indices[j]. |
© 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/r1.15/api_docs/python/tf/contrib/image/bipartite_match