我目前正在使用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(静态图)一起使用?