如何使用Tensorflow中的label_image.py一次测试多个图像

时间:2017-09-28 16:30:03

标签: python machine-learning tensorflow computer-vision artificial-intelligence

我一直在研究一种方法来加载我的多分类张量流模块,一次扫描25-30张图像。它目前在测试阶段一次只运行一个图像。我们的想法是通过编辑的label_image.py一次性比较它们,并通过tensorflow库在docker中运行。我有希望自己解决这个问题,但无济于事。我目前的假设是虽然可以同时测试多个图像,但我不确定我应该在labe_image.py中编辑哪些内容。我提前谢谢了。

1 个答案:

答案 0 :(得分:1)

你可以在

之后使用for循环
graph = load_graph(model_file)
files=["tf_files/1.jpg","tf_files/2.jpg"]
像这样:

for file_name in files:
  t = read_tensor_from_image_file(file_name,input_height=input_height,input_std=input_std)