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

Question

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

3

我目前正在使用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（静态图）一起使用？

- Bastian

1个回答

网页内容由stack overflow 提供, 点击上面的

可以查看英文原文，
原文链接

- Panfeng Li · Accepted Answer

一个简单的方法是直接使用Python的imageio，然后使用tf.py_func进行封装。