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
打包。
我目前正在使用 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
打包。