将图片提供给模型张量流进行训练

时间:2016-12-12 23:55:30

标签: python tensorflow

我试图将图片导入我的模型进行培训,就像示例" image_retraining " : https://github.com/tensorflow/tensorflow/tree/master/tensorflow/examples/image_retraining 但导入过程图像位于 classify_image_graph_def.pb 中,所以我不知道它是如何工作的

我找到了这些方法:
Tensorflow multithreading image loading

https://stackoverflow.com/a/36947632/7040568

https://github.com/samjabrahams/tensorflow-workshop/blob/master/04%20Feedforward%20Network.ipynb

你介意告诉我导入训练模型图片的标准张量流方法是什么? 我还没有使用tf.train.batchtf.train.Coordinatortf.train.start_queue_runners

我真的很困惑,因为我还在学习如何使用张量流,所以我只需要简单或标准的方式

谢谢你和问候,

1 个答案:

答案 0 :(得分:0)

一种方法是使用PIL或OpenCV自行加载它们,例如cv2.imread(filename.png),然后将其作为尺寸为[feed_dict的数组提供给您的图表(number of images (batch size), height, width, channels (3 if RGB)])。