Tensorflow 2中tf.contrib.image.bipartite_match的等价物是什么？

Question

由于TensorFlow从TF2中删除了contrib库，有人知道如何移植这个特殊的函数吗？

Answer 1

我需要查看日志才能帮助您，但请尝试安装"tensorflow_addons“

将tensorflow_addons作为tfa导入

tfa.image..

有关更多信息，请访问https://www.tensorflow.org/addons/api_docs/python/tfa/image/transform

另一种选择是使用tf-image。(https://pypi.org/project/tf-image/)或：

从tensorflow.python.ops导入image_ops

mage_ops.bipartite_match(distance_matrix，num_valid_rows)

Answer 2

我们按照https://www.tensorflow.org/guide/create_op中的步骤并使用https://github.com/tensorflow/tensorflow/commit/1244c8fdda9efb11a5386dce40fbb885f0df190b中的原始代码编写了一个自定义的C++ op。性能与TF -1 tf.contrib.image.bipartite_match相当。