Ich bin sehr neu bei TensorFlow. Ich mache dasbildklassifizierung mit meiner eigenen Trainingsdatenbank.

Nachdem ich meinen eigenen Datensatz trainiert habe, habe ich keine Ahnung, wie ich das Eingabebild klassifizieren soll.

Hier ist mein Code für meinen eigenen Datensatz vorbereiten

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)

Ist dies ein korrekter Code zum Trainieren des Datensatzes?

Anschließend versuche ich, die eingegebenen Bilder mit dem folgenden Code zu klassifizieren.

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)

Es wird jedoch nicht die vorhergesagte Bezeichnung für die eingegebenen Bilder zurückgegeben. Ich suche jemanden, der mir beibringt, wie man die Bilder anhand meines eigenen Datensatzes klassifiziert.

Vielen Dank