Tensorflow：如何使用Tensorflow的数据集API来使用TIFF图像？

Question

我目前正在使用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。现在，我实现了一个类，该类将从TIFF容器中提取特定页面作为Numpy数组。是否可以通过自定义图像加载器（例如类）将此API（静态图）使用？

Answer 1

一种简单的方法是直接使用python imageio，然后使用tf.py_func对其进行打包。