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

问题描述

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

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

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

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

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

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

解决方法

暂无找到可以解决该程序问题的有效方法,小编努力寻找整理中!

如果你已经找到好的解决方法,欢迎将解决方案带上本链接一起发送给小编。

小编邮箱:dio#foxmail.com (将#修改为@)