我们正在使用Tensorflow对图像中的对象进行分类。为此，我们使用通用的对象识别模型。由于模型速度有点慢，我想知道是否可以仅使用模型的几个类？
我正在寻找的东西是这样的：
# Loading the model
detector = hub.load(model_path).signatures['default']
# subsetting it
subset_detector=detector["classes"]["cars"]
我很确定实际的解决方案会更复杂。我也可以接受模型的重新训练。只要它变得更快。
我记得前一段时间研究了这个问题。我还记得找到答案。现在，我试图再次找到该答案。我重读了savedModel documentation和所有相关文档。但是我找不到关于此的任何信息。我希望更多使用tensorflow的人可以帮助我。