Skip to main content
 首页 » 编程设计

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

2025年02月15日24Renyi-Fan

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

请您参考如下方法:

你可以在之后使用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)