如何在Tensorflow 2中实现随机剪切预处理层?

时间:2020-08-11 00:10:40

标签: python tensorflow keras

我正在寻找一个随机剪切一批图像的层,例如tf.keras.layers.experimental.preprocessing中的预处理层。但是,似乎没有任何这样的层。

在TF中实现剪切层存在一些类似的问题,但是在tf.contribherehere)中使用了不赞成使用的方法。有人可以指出我在张量流2中实施随机剪切的方向吗?

1 个答案:

答案 0 :(得分:1)

Tensorflow 2.x 中 tf.contrib 的一些操作移至 Tensorflow 插件。 tf.contrib.image.transoform 的等价物是 tfa.image.transform 有关图书馆的更多详细信息,请查找 here

同时使用 tf.keras.preprocessing.image.random_shear

看看 Random_shear