Tensorflow:如何使用 Tensorflow 的 Dataset API 来使用 TIFF 图像?

Tensorflow: How to use Tensorflow's Dataset API to use TIFF images?

我目前正在使用 Tensorflow 的 Dataset API 加载和预处理图像以进行训练和测试。这是我用来执行此操作的代码片段:

 image_string = tf.read_file(self._data_conf.image_dir + in_image)
 label_string = tf.read_file(self._data_conf.label_dir + in_label)
 image = tf.image.decode_png(image_string, channels=self._num_channels)
 label = tf.image.decode_png(label_string, channels=self._num_channels)

问题是,Tensorflow 不支持 TIFF 容器。这里只支持函数tf.image.decode_image、tf.image.decode_jpeg和tf.image.decode_png。现在,我已经实现了一个 class,它从 TIFF 容器中提取特定页面作为 Numpy 数组。有没有办法将此 API(静态图)与自定义图像加载器一起使用,例如 class?

一个简单的方法就是直接使用python imageio 然后使用tf.py_func 打包。