Soy muy nuevo en TensorFlow. Estoy haciendo elclasificación de imágenes usando mi propia base de datos de entrenamiento.

Sin embargo, después de entrenar mi propio conjunto de datos, no tengo idea de cómo clasificar la imagen de entrada.

Aquí está mi código parapreparando mi propio conjunto de datos

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)

¿Es este un código correcto para entrenar el conjunto de datos?

Luego, trato de usarlo para clasificar las imágenes de entrada con el siguiente 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)

Sin embargo, no devuelve la etiqueta predicha para las imágenes de entrada. Estoy buscando a alguien que me enseñe cómo clasificar las imágenes utilizando mi propio conjunto de datos.

Gracias.