我一直在研究一种方法来加载我的多分类 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)
