Eu sou muito novo no TensorFlow. Eu estou fazendo oclassificação da imagem usando meu próprio banco de dados de treinamento.

No entanto, depois de treinar meu próprio conjunto de dados, não tenho idéia de como classificar a imagem de entrada.

Aqui está o meu código parapreparando meu próprio conjunto de dados

filenames = ['01.jpg', '02.jpg', '03.jpg', '04.jpg']
label = [0,1,1,1]
filename_queue = tf.train.string_input_producer(filenames)

reader = tf.WholeFileReader()
filename, content = reader.read(filename_queue)
image = tf.image.decode_jpeg(content, channels=3)
image = tf.cast(image, tf.float32)
resized_image = tf.image.resize_images(image, 224, 224)

image_batch , label_batch= tf.train.batch([resized_image,label], batch_size=8, num_threads = 3, capacity=5000)

Esse é um código correto para treinar o conjunto de dados?

Depois, tento usá-lo para classificar as imagens de entrada com o seguinte código.

test = ['test.jpg', 'test2.jpg']
test_queue=tf.train.string_input_producer(test)
reader = tf.WholeFileReader()
testname, test_content = reader.read(test_queue)
test = tf.image.decode_jpeg(test_content, channels=3)
test = tf.cast(test, tf.float32)
resized_image = tf.image.resize_images(test, 224,224)

with tf.Session() as sess:
    coord = tf.train.Coordinator()
    threads = tf.train.start_queue_runners(coord=coord)
    res = sess.run(resized_image)
    coord.request_stop()
    coord.join(threads)

No entanto, ele não retorna o rótulo previsto para as imagens de entrada. Estou procurando alguém para me ensinar como classificar as imagens usando meu próprio conjunto de dados.

Obrigado.