Tensorflow 中的图像变形

Posted 2023-03-29

【中文标题】Tensorflow 中的图像变形

【英文标题】：Image warping in Tensorflow

【发布时间】：2017-09-15 15:30:08

【问题描述】：

我正在处理视频序列和光流。我想知道，Tensorflow 是否有扭曲图像的操作。模拟 Torch 中的 image.warp https://github.com/torch/image/blob/master/doc/paramtransform.md

如果没有内置这样的操作，也许有开源代码。或者你可以提供指针让我在 TF 中实现这个操作。

谢谢！

【参考方案1】：

我还没有找到内置函数。但是answer to this question 可能会对您有所帮助。

它使用标准 Tensorflow Ops 构建并执行 bilinear interpolation，但与真正的 CUDA 优化 Ops 相比，我想它不会很快。您还需要为批次、彩色图像和所需的填充类型扩展它。

【参考方案2】：

是的。在tf2 中移动到TensorFlow Addons (here)：

@tf.function
tfa.image.dense_image_warp(
    image: tfa.types.TensorLike,
    flow: tfa.types.TensorLike,
    name: Optional[str] = None
) -> tf.Tensor

【参考方案3】：

opencv有这些功能：warpPerspective(),perspectiveTransform()