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图像。 （https://pypi.org/project/tf-image/）或：

从tensorflow.python.ops导入image_ops

mage_ops.bipartite_match（distance_matrix，num_valid_rows）