Como você carrega, rotula e alimenta dados jpeg no Tensorflow?
Estou tentando alimentar imagens 1750 * 1750 no Tensorflow, mas não sei como rotular e alimentar os dados após converter as imagens em um Tensor usando a função tf.image.decode_jpeg ().
Atualmente, meu código é:
import tensorflow as tf
import numpy as np
import imageflow
import os, glob
sess = tf.InteractiveSession()
def read_jpeg(filename_queue):
reader = tf.WholeFileReader()
key, value = reader.read(filename_queue)
my_img = tf.image.decode_jpeg(value)
my_img.set_shape([1750, 1750, 1])
print(value)
return my_img
#####################################################
def read_image_data():
jpeg_files = []
images_tensor = []
i = 1
WORKING_PATH = "/Users/Zanhuang/Desktop/NNP/DATA"
jpeg_files_path = glob.glob(os.path.join(WORKING_PATH, '*.jpeg'))
for filename in jpeg_files_path:
print(i)
i += 1
jpeg_files.append(filename)
filename_queue = tf.train.string_input_producer(jpeg_files)
mlist = [read_jpeg(filename_queue) for _ in range(len(jpeg_files))]
init = tf.initialize_all_variables()
sess = tf.Session()
sess.run(init)
images_tensor = tf.convert_to_tensor(images_tensor)
sess.close()
Agora, como disse anteriormente, preciso alimentar e rotular os dados. Eu vi os arquivos do tutorial do CIFAR-10, mas eles armazenaram os rótulos em um arquivo e pretendo não fazê-lo.
Eu sou bastante novo no Tensorflow, portanto, mantenha a resposta o mais detalhada possível.
Obrigado!