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

Question

我一直在研究一种方法来加载我的多分类 tensorflow 模块，一次扫描 25-30 个图像。它目前在测试阶段一次只能运行一个图像。这个想法是通过编辑的 label_image.py 一次比较它们，并通过 tensorflow 库在 docker 中运行。我曾希望自己解决这个问题，但无济于事。我目前假设虽然可以一次测试多个图像，但我不确定我应该在 labe_image.py 中的什么地方进行编辑。提前致谢。

Answer 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)