TPU支持使用变换进行图像增强

Question

我最近使用tf函数和tf记录重新编码了我的输入图像管道，这样我就可以使用TPU(以前使用opencv的自定义tf.py_function )。问题是，我使用tfa.image.transform来应用旋转和缩放增强，但TPU显然不支持tensorflow插件包。在TPU管道中使用变换执行图像增强有哪些替代方案？

Answer 1

我用以下代码替换了tfa.image.transform，从而解决了这个问题：

from tensorflow.python.keras.layers.preprocessing import image_preprocessing as image_ops
image_ops.transform()

事实证明，tfa.image.transform只是这个函数的一个包装器。这两个函数的参数是相同的。