将.npy输入到tensorflow数据管道中

Question

当输入的是jpg图片时，我的源代码不起作用。

但是现在输入更改为具有shape[224,224,4]的.npy文件。 函数decode_jpeg不起作用。

这是代码：

    image_content = tf.read_file(file_queue[0])
    image_data = tf.image.decode_jpeg(image_content, channels=3) 
    img = tf.image.resize_images(image_data, [224, 224])
    labels = tf.one_hot(file_queue[1],self.num_class, dtype=tf.uint8)       
    image_batch, label_batch = tf.train.shuffle_batch([img,labels],batch_size=self.batch_size,capacity=2000,min_after_dequeue=1000)

Answer 1

Numpy的.npy文件包含已解码的图像数据，您无法直接使用tensorflow读取它们。它需要在Tensorflow中支持Numpy二进制格式。

考虑使用python代码阅读它们。如果您使用的是from_generator，则可以使用Dataset API方法；如果您使用的是queues API，则可以手动将它们输入队列。