我是Tensorflow的初学者,我想玩一下对象检测api。我从Tensorflow Hub下载了经过预训练的模型,并解压缩了.tar.gz。
这部分代码现在如下所示:
hub_model = tf.saved_model.load(filepath)
# filepath is the directory containing saved_model.pb
稍后在代码中,我实际上是对某些图像进行推理并绘制一些框,并且一切工作正常。唯一的问题是,加载模型实际上要花2分钟,因此测试代码中的任何更改都需要2分钟。
我应该以某种方式保持模型加载,还是我缺少/做错了什么?
如果有人可以提供解决方案/向我指出正确的方向,那就太棒了!