如何在单个 tensorflow 会话中运行多个分类模型（keras、caffe 等）？答案

【问题标题】：How to run multiple classification models (keras, caffe, etc) in a single tensorflow session?如何在单个 tensorflow 会话中运行多个分类模型（keras、caffe 等）？
【发布时间】：2020-06-15 22:23:43
【问题描述】：

我正在尝试运行对象检测模型 (SSD+Inceptionv2) 并获得边界框作为结果。从帧中获取边界框后，裁剪后的图像应该通过不同的分类模型（在 keras 和 caffe 等不同库上进行训练）并获得预测结果。

在这里，我在 tensorflow 会话中从对象检测模型中获取边界框，当尝试使用任何其他模型预测方法时，该预测方法默认需要 tensorflow 对象检测输入。由于这种分类模型预测方法不起作用。

知道如何在对象检测输出边界框（裁剪图像）上运行不同的分类模型吗？

【问题讨论】：

标签： python tensorflow keras deep-learning caffe

【解决方案1】：

会话链接到特定的 Tensorflow Graph 实例。如果您想为所有人提供一个会话，则需要将所有模型放在同一个图中。这可能会导致您为张量命名问题，并且 IMO 通常是一个坏主意（您应该将彼此不相关的事物分开）。

将所有内容放在同一个图中也会提高模型的资源需求（即使只运行一个子图，您也总是会加载所有内容），这是将事物拆分为独立图中的另一个原因。使用独立图表，您将不得不使用多个会话。

Please refer to this

【讨论】：