如何在TensorFlow 2.3中更快地加载预训练模型?

时间:2020-10-14 03:45:58

标签: tensorflow object-detection-api

我是Tensorflow的初学者,我想玩一下对象检测api。我从Tensorflow Hub下载了经过预训练的模型,并解压缩了.tar.gz。

这部分代码现在如下所示:

hub_model = tf.saved_model.load(filepath)
# filepath is the directory containing saved_model.pb

稍后在代码中,我实际上是对某些图像进行推理并绘制一些框,并且一切工作正常。唯一的问题是,加载模型实际上要花2分钟,因此测试代码中的任何更改都需要2分钟。

我应该以某种方式保持模型加载,还是我缺少/做错了什么?

如果有人可以提供解决方案/向我指出正确的方向,那就太棒了!

0 个答案:

没有答案