Tensorflow:模型的Jpeg输入

时间:2017-02-12 20:13:53

标签: python tensorflow jpeg

我目前正在使用转移学习retrain.py文件,并且我使用预训练的重量制作了基本的VGG16模型。我在tensorboard中查看初始v3图,看看它如何接受不同值的jpeg图像。 Inception V3 input我完成了每个操作的内容,我认为我已正确设置,但显然没有。

self.input_image = tf.placeholder(tf.string, shape=[])
self.decode_jpeg = tf.image.decode_jpeg(self.input_image, channels=3, ratio = 1, fancy_upscaling=True, try_recover_truncated=False)
self.cast = tf.cast(self.decode_jpeg, tf.float32)
self.multi = tf.expand_dims(self.cast, 0)
self.sized = tf.image.resize_bilinear(self.multi, [224, 224], align_corners=False)

这是我用来测试它的。

prob = sess.run(vgg16.output, feed_dict={vgg16.input_image: 'laska.jpg'})

所以我只是在文件夹中传递一串jpg的名字。它告诉我......

ValueError: Cannot feed value of shape (1,) for Tensor 'Placeholder_1:0', which has shape '()'

那么我做错了什么?我该怎么办才能将我的模型移植到retrain.py文件中?

0 个答案:

没有答案