Tensorflow 2.0 和图像处理

Question

我只是将一些代码转换为 tensorflow v2，但我遇到了一个问题。 我曾经在 tensorflow v1 中使用一些 tf.contrib.image 函数（dense_image_warp 和旋转）。我不知道如何在 v2 中替换它们。 这些函数不是 tf.image 模块的一部分。

tf_upgrade_v2 脚本无法转换这些方法。

有人知道如何替换它们以兼容 v2 吗？

谢谢

Answer 1

许多以前位于 tf.contrib 中的图像处理功能已迁移到 TensorFlow 2.0 中的其他位置。 dense_image_warp 和 rotate 现在是 TensorFlow Addons 中 tfa.image 模块的一部分。