2016-10-28 6 views
2

TensorFlowでトレーニング用に自分の画像を読み込もうとしました。ただし、エラーがあるようです:TensorFlowでのトレーニングのための画像のサイズ変更

ValueError: 'size' must be a 1-D Tensor of 2 elements: new_height, new_width. 

次のコードサンプルで何が問題になっていますか?

filenames=['images/000001.jpg','images/000002.jpg','images/000003.jpg','images/000004.jpg'] 
labels=[1,0,1,0] 

filename_queue=tf.train.string_input_producer(filenames) 

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

image_batch, label_batch=tf.train.batch([resized_images, labels], batch_size=2) 
+1

エラーが何であるの第2引数にすることですか?それを提供してください。 – Mox

答えて

6

答えはtf.image.resize_imagesテンソル

resized_images=tf.image.resize_images(images, [224, 224]) 
関連する問題