Bonjour ,


Je débute dans Tensorflow. J'essaie donc de créer un set d'images de chats et de chiens de tailles différentes (10000 images de chaque "type" d'animal) pour l'entrainement. Mon code ne renvoie pas d'erreur mais mon batch ne se créé pas.
peut-être pourriez vous m'aider ?
j'ai fais plusieurs tests sans succès.

import tensorflow as tf #tensor flow
import numpy as np
import glob, os
 
 
from pprint import pprint
 
 
filenames = list()
 
os.chdir(".")
 for file in glob.glob("*.jpg"):
     filenames.append( file)
 
#filenames = ['0.jpg','1.jpg','2.jpg','3.jpg']
 
 
 
# filenames = ['im_01.jpg', 'im_02.jpg', 'im_03.jpg', 'im_04.jpg']
 
 
filename_queue = tf.train.string_input_producer(filenames)
 
 
reader = tf.WholeFileReader()
filename, content = reader.read(filename_queue)
image = tf.image.decode_jpeg(filename, channels=3)
image = tf.cast(image, tf.float32)
resized_image = tf.image.resize_images(image, [224, 224])
 
labels=[1,0,1,0] 
image_batch, label_batch=tf.train.batch([resized_image, labels], batch_size=2)
#image_batch = tf.train.batch([resized_image], batch_size=64)
 
print (image_batch)

Et en output du print:

Tensor("batch:0", shape=(2, 224, 224, 3), dtype=float32)

Merci de votre aide
Thomas

Bonjour,

Tensorflow ne fonctionne pas comme un code standard. La ce que tu as fais, c'est le graph. Il faut ensuite le faire tourner sur une session.

https://www.tensorflow.org/api_docs/python/tf/Session

Bon courage,